8 def inference(self, det_bboxes, det_labels, frame_id, num_classes=1):
9 if frame_id == 0:
10 self.tracker.reset()
11
12 track_bboxes, track_labels, track_ids = self.tracker.track(
13 img = None,
14 img_metas = [dict(frame_id=frame_id)],
15 model = self,
16 bboxes=det_bboxes,
17 labels=det_labels,
18 frame_id=frame_id)
19
20 track_results = outs2results(
21 bboxes=track_bboxes,
22 labels=track_labels,
23 ids=track_ids,
24 num_classes=num_classes)
25 det_results = outs2results(
26 bboxes=det_bboxes, labels=det_labels, num_classes=num_classes)
27
28 return dict(
29 det_bboxes=det_results['bbox_results'],
30 track_bboxes=track_results['bbox_results'])
