Inheritance diagram for byte_tracker_mod_runner.ByteTrackerModRunner:
Public Member Functions | |
| __init__ (self, tracker=None, motion=None) | |
| run_tracker (self, image, mask, det_bboxes, det_labels, frame_id, num_classes=1) | |
| get_tracks (self) | |
Public Attributes | |
| motion | |
| tracker | |
Detailed Description
Definition at line 10 of file byte_tracker_mod_runner.py.
Constructor & Destructor Documentation
◆ __init__()
| byte_tracker_mod_runner.ByteTrackerModRunner.__init__ | ( | self, | |
| tracker = None, | |||
| motion = None ) |
Definition at line 11 of file byte_tracker_mod_runner.py.
11 def __init__(self, tracker=None, motion=None):
12 # super().__init__(tracker=tracker, motion=motion)
13 self.motion = OpticalFlowMotion()
14 self.tracker = ByteTrackerMod()
15
Member Function Documentation
◆ get_tracks()
| byte_tracker_mod_runner.ByteTrackerModRunner.get_tracks | ( | self | ) |
Definition at line 37 of file byte_tracker_mod_runner.py.
37 def get_tracks(self):
38 track_ids = []
39 track_bboxes = np.zeros((0, 4))
40 for track_id in self.tracker.tracks.keys():
41 if not self.tracker.tracks[track_id].valid:
42 continue
43
44 track_bboxes = np.concatenate(
45 (track_bboxes, self.tracker.tracks[track_id].mean[:4][None]), axis=0)
46 track_ids.append(track_id)
47
48 return track_bboxes, track_ids
◆ run_tracker()
| byte_tracker_mod_runner.ByteTrackerModRunner.run_tracker | ( | self, | |
| image, | |||
| mask, | |||
| det_bboxes, | |||
| det_labels, | |||
| frame_id, | |||
| num_classes = 1 ) |
Definition at line 16 of file byte_tracker_mod_runner.py.
16 def run_tracker(self, image, mask, det_bboxes, det_labels, frame_id, num_classes=1):
17 track_bboxes, track_labels, track_ids = self.tracker.track(
18 img=image,
19 mask=mask,
20 motion=self.motion,
21 bboxes=det_bboxes,
22 labels=det_labels,
23 frame_id=frame_id)
24
25 track_results = outs2results(
26 bboxes=track_bboxes,
27 labels=track_labels,
28 ids=track_ids,
29 num_classes=num_classes)
30 det_results = outs2results(
31 bboxes=det_bboxes, labels=det_labels, num_classes=num_classes)
32
33 return dict(
34 det_bboxes=det_results['bbox_results'],
35 track_bboxes=track_results['bbox_results'])
36
Member Data Documentation
◆ motion
| byte_tracker_mod_runner.ByteTrackerModRunner.motion |
Definition at line 13 of file byte_tracker_mod_runner.py.
◆ tracker
| byte_tracker_mod_runner.ByteTrackerModRunner.tracker |
Definition at line 14 of file byte_tracker_mod_runner.py.
The documentation for this class was generated from the following file:
- smtrack/runner/byte_tracker_mod_runner.py
