minor fixes

Author: NirAharon

.gitignore

@@ -9,3 +9,4 @@ wandb/
 *egg-info*
 fast_reid/datasets/MOT17-ReID
 */__pycache__
+*.so

tools/mc_demo.py

@@ -47,7 +47,6 @@ def make_parser():
     parser.add_argument("--new_track_thresh", default=0.7, type=float, help="new track thresh")
     parser.add_argument("--track_buffer", type=int, default=30, help="the frames for keep lost tracks")
     parser.add_argument("--match_thresh", type=float, default=0.8, help="matching threshold for tracking")
-    parser.add_argument("--aspect_ratio_thresh", type=float, default=1.6, help="threshold for filtering out boxes of which aspect ratio are above the given value.")
     parser.add_argument('--min_box_area', type=float, default=10, help='filter out tiny boxes')
     parser.add_argument("--fuse-score", dest="fuse_score", default=False, action="store_true", help="fuse score and iou for association")
 
@@ -179,8 +178,7 @@ def image_demo(predictor, vis_folder, current_time, args):
         for t in online_targets:
             tlwh = t.tlwh
             tid = t.track_id
-            vertical = tlwh[2] / tlwh[3] > args.aspect_ratio_thresh
-            if tlwh[2] * tlwh[3] > args.min_box_area and not vertical:
+            if tlwh[2] * tlwh[3] > args.min_box_area:
                 online_tlwhs.append(tlwh)
                 online_ids.append(tid)
                 online_scores.append(t.score)
@@ -263,8 +261,7 @@ def imageflow_demo(predictor, vis_folder, current_time, args):
             for t in online_targets:
                 tlwh = t.tlwh
                 tid = t.track_id
-                vertical = tlwh[2] / tlwh[3] > args.aspect_ratio_thresh
-                if tlwh[2] * tlwh[3] > args.min_box_area and not vertical:
+                if tlwh[2] * tlwh[3] > args.min_box_area:
                     online_tlwhs.append(tlwh)
                     online_ids.append(tid)
                     online_scores.append(t.score)

tools/mc_demo_yolov7.py

@@ -135,8 +135,7 @@ def detect(save_img=False):
                 tlbr = t.tlbr
                 tid = t.track_id
                 tcls = t.cls
-                vertical = tlwh[2] / tlwh[3] > opt.aspect_ratio_thresh
-                if tlwh[2] * tlwh[3] > opt.min_box_area and not vertical:
+                if tlwh[2] * tlwh[3] > opt.min_box_area:
                     online_tlwhs.append(tlwh)
                     online_ids.append(tid)
                     online_scores.append(t.score)
@@ -214,7 +213,7 @@ def detect(save_img=False):
 
     # tracking args
     parser.add_argument("--track_high_thresh", type=float, default=0.3, help="tracking confidence threshold")
-    parser.add_argument("--track_low_thresh", default=0.1, type=float, help="lowest detection threshold")
+    parser.add_argument("--track_low_thresh", default=0.05, type=float, help="lowest detection threshold")
     parser.add_argument("--new_track_thresh", default=0.4, type=float, help="new track thresh")
     parser.add_argument("--track_buffer", type=int, default=30, help="the frames for keep lost tracks")
     parser.add_argument("--match_thresh", type=float, default=0.7, help="matching threshold for tracking")
@@ -239,6 +238,7 @@ def detect(save_img=False):
                         help='threshold for rejecting low appearance similarity reid matches')
 
     opt = parser.parse_args()
+
     opt.jde = False
     opt.ablation = False
 

tracker/mc_bot_sort.py

@@ -293,7 +293,7 @@ def update(self, output_results, img):
 
         if len(dets) > 0:
             '''Detections'''
-            if self.args.with_reid or self.args.jde:
+            if self.args.with_reid:
                 detections = [STrack(STrack.tlbr_to_tlwh(tlbr), s, c, f) for
                               (tlbr, s, c, f) in zip(dets, scores_keep, classes_keep, features_keep)]
             else: