fix bug of deepsort and c_biou

Author: JackWoo0831

tracker/c_biou_tracker.py

@@ -146,6 +146,12 @@ def update(self, new_track, frame_id):
             self.motion_state1 = self.get_buffer_bbox(level=1, bbox=new_tlwh)
             self.motion_state2 = self.get_buffer_bbox(level=2, bbox=new_tlwh)
 
+        # update status
+        self.state = TrackState.Tracked
+        self.is_activated = True
+
+        self.time_since_update = 0
+
     @staticmethod
     def xywh2tlbr(xywh):
         """

tracker/deepsort.py

@@ -190,8 +190,8 @@ def update(self, det_results, ori_img):
             lost_stracks.append(track)
 
         # deal with unconfirmed tracks, match new track of last frame and new high conf det
-        matched_pair2, u_tracks2_idx, u_det2_idx = matching.matching_cascade(self.gated_metric, 0.7, self.max_time_lost, 
-                                                                unconfirmed, u_det1)
+        IoU_dist = matching.iou_distance(atracks=unconfirmed, btracks=u_det1)
+        matched_pair2, u_tracks2_idx, u_det2_idx = matching.linear_assignment(IoU_dist, thresh=0.9)
 
         for itrack_match, idet_match in matched_pair2:
             track = unconfirmed[itrack_match]

tracker/track.py

@@ -184,7 +184,7 @@ def main(opts, cfgs):
             timer.toc()  # end timing this image
 
             if opts.save_images:
-                plot_img(img0, frame_id, [cur_tlwh, cur_id, cur_cls], save_dir=os.path.join(DATASET_ROOT, 'reuslt_images', seq))
+                plot_img(img0, frame_id, [cur_tlwh, cur_id, cur_cls], save_dir=os.path.join(DATASET_ROOT, 'result_images', seq))
 
             frame_id += 1
 
@@ -306,7 +306,7 @@ def save_videos(seq_names):
         seq_names = [seq_names]
 
     for seq in seq_names:
-        images_path = os.path.join(DATASET_ROOT, 'reuslt_images', seq)
+        images_path = os.path.join(DATASET_ROOT, 'result_images', seq)
         images_name = sorted(os.listdir(images_path))
 
         to_video_path = os.path.join(images_path, '../', seq + '.mp4')