cecile-supercool-tracker 0.0.1

Machine learning framework for building object trackers and similarity search engines
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

from similari import Universal2DBoxKalmanFilter, BoundingBox

if __name__ == '__main__':
    f = Universal2DBoxKalmanFilter()
    state = f.initiate(BoundingBox(0.0, 0.0, 5.0, 10.0).as_xyaah())
    state = f.predict(state)
    box_ltwh = state.bbox()
    box_xyaah = state.universal_bbox()
    print(box_ltwh)
    print(box_xyaah)
    # if work with oriented box
    # import Universal2DBox and use it
    #
    #box_xyaah = state.universal_bbox()
    #print(box_xyaah)

    state = f.update(state, BoundingBox(0.2, 0.2, 5.1, 9.9).as_xyaah())
    state = f.predict(state)
    box_ltwh = state.bbox()
    print(box_ltwh)

    for i in range(1, 21):
        state = f.predict(state)
        box_xyaah = state.universal_bbox()
        print("Prediction:", box_xyaah)

        obs = BoundingBox(0.2 + i * 0.2, 0.2 + i * 0.2, 5.0, 10.0).as_xyaah()
        state = f.update(state, obs)
        print("Observation:", obs)