fix some bugs of demo and sort

Author: JackWoo0831

README.md

@@ -21,6 +21,7 @@ git checkout v2  # change to v2 branch !!
 
 ## 🗺️ Latest News
 
+- ***2025.4.14*** Fix some minor bugs [issue#144](https://github.com/JackWoo0831/Yolov7-tracker/issues/144), and fix the lost tracklet bugs in sort.
 - ***2025.4.7*** Add more Re-ID modules (ShuffleNet, VehicleNet, MobileNet), fix some bugs (such as abandon bbox aspect ratio updating if the tracklet is not activated), and add some functions (customized low filter threshold, fuse detection score, etc.)
 - ***2025.4.3*** Support the newest ultralytics version (YOLO v3 ~ v12) and fix some bugs of hybrid sort.
 

README_CN.md

@@ -15,6 +15,7 @@ git checkout v2  # change to v2 branch !!
 
 ## 🗺️ 最近更新
 
+- ***2025.4.14*** 修复[issue#144](https://github.com/JackWoo0831/Yolov7-tracker/issues/144)中提到的一些bug，修复sort对丢失轨迹处理的bug.
 - ***2025.4.7*** 增加更多Re-ID模型 (ShuffleNet, VehicleNet, MobileNet), 修复一些bug (例如在轨迹为非活动状态时停止更新边界框长宽), 增加一些小功能 (例如可以修改两阶段关联策略的最低阈值，原来是固定的0.1; 增加将IoU和检测置信度融合的选项)
 - ***2025.4.3*** 增加了ultralytics库最新版本的支持，修复了hybrid sort中的一些bug.
 

tracker/track_demo.py

@@ -134,6 +134,11 @@ def main(args):
     if args.save_videos:
         args.save_images = True
 
+    save_dir = args.save_dir
+    save_obj_name = args.obj.replace('/', '_')  # save seq name w.r.t. the obj name, but replace '/'
+    logger.info(f'demo result will be saved in {os.path.join(save_dir), save_obj_name}.txt')
+    logger.info(f"images and videos (if you enable it) will be saved in {os.path.join(save_dir, save_obj_name, 'vis_results')}")
+
     """2. load detector"""
     device = select_device(args.device)
 
@@ -196,9 +201,6 @@ def main(args):
 
     tracker = TRACKER_DICT[args.tracker](args, )
 
-
-    save_dir = args.save_dir
-
     process_bar = enumerate(data_loader)
     process_bar = tqdm(process_bar, total=len(data_loader), ncols=150)
 
@@ -219,7 +221,7 @@ def main(args):
         # get detector output 
         with torch.no_grad():
             if 'ultra' in args.detector:
-                output = model.predict(img, conf=args.conf_thresh, iou=args.nms_thresh)
+                    output = model.predict(img, conf=args.conf_thresh, iou=args.nms_thresh, verbose=False)
             else:
                 output = model(img)
 
@@ -264,14 +266,14 @@ def main(args):
 
         if args.save_images:
             plot_img(img=ori_img, frame_id=frame_idx, results=[cur_tlwh, cur_id, cur_cls], 
-                        save_dir=os.path.join(save_dir, 'vis_results'))
+                        save_dir=os.path.join(save_dir, save_obj_name, 'vis_results'))
 
-    save_results(folder_name=os.path.join(save_dir, 'txt_results'), 
-                    seq_name='demo', 
-                    results=results)
+    save_results(save_dir=save_dir, 
+                seq_name=save_obj_name,  
+                results=results)
 
     if args.save_videos:
-        save_video(images_path=os.path.join(save_dir, 'vis_results'))
+        save_video(images_path=os.path.join(save_dir, save_obj_name, 'vis_results'))
         logger.info(f'save video done')
 
 if __name__ == '__main__':

tracker/tracker_dataloader.py

@@ -185,6 +185,8 @@ def __init__(self, file_name, img_size=[640, 640], model='yolox', legacy_yolox=T
 
         self.legacy = legacy_yolox
 
+        self.other_param = kwargs
+
     def __getitem__(self, idx):
 
         if not self.is_video:

tracker/trackers/deepsort_tracker.py

@@ -32,7 +32,7 @@ def __init__(self, args, frame_rate=30):
 
         self.motion = args.kalman_format
 
-        self.with_reid = args.reid
+        self.with_reid = True  # In deepsort, reid model must be included
 
         self.reid_model = None
         if self.with_reid:

tracker/trackers/sort_tracker.py

@@ -86,6 +86,12 @@ def update(self, output_results, img, ori_img):
                 track.re_activate(det, self.frame_id, new_id=False)
                 refind_tracklets.append(track)
 
+        for it in u_track:
+            track = tracklet_pool[it]
+            if not track.state == TrackState.Lost:
+                track.mark_lost()
+                lost_tracklets.append(track)
+
 
         '''Deal with unconfirmed tracks, usually tracks with only one beginning frame'''
         detections = [detections[i] for i in u_detection]