| import numpy as np |
| from collections import OrderedDict |
|
|
|
|
| class TrackState(object): |
| New = 0 |
| Tracked = 1 |
| Lost = 2 |
| Removed = 3 |
|
|
|
|
| class BaseTrack(object): |
| _count = 0 |
|
|
| track_id = 0 |
| is_activated = False |
| state = TrackState.New |
|
|
| history = OrderedDict() |
| features = [] |
| curr_feature = None |
| score = 0 |
| start_frame = 0 |
| frame_id = 0 |
| time_since_update = 0 |
|
|
| |
| location = (np.inf, np.inf) |
|
|
| @property |
| def end_frame(self): |
| return self.frame_id |
|
|
| @staticmethod |
| def next_id(): |
| BaseTrack._count += 1 |
| return BaseTrack._count |
|
|
| def activate(self, *args): |
| raise NotImplementedError |
|
|
| def predict(self): |
| raise NotImplementedError |
|
|
| def update(self, *args, **kwargs): |
| raise NotImplementedError |
|
|
| def mark_lost(self): |
| self.state = TrackState.Lost |
|
|
| def mark_removed(self): |
| self.state = TrackState.Removed |
