tracking types 2

scripts/track_detection.py

@@ -101,7 +101,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
                 image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
 
 
-    if False:
+    if config['associativity'] == "bytetrack":
 
         low_outputs = do_suppression(inference, conf_thres=config['low_conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
         low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
@@ -113,7 +113,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
     else:
         outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 
-        if config['use_associative']:
+        if config['associativity'] == "boost":
 
             do_confidence_boost(inference, outputs, boost_power=config['boost_power'], boost_decay=config['boost_decay'], verbose=verbose)
 

scripts/track_eval.py

@@ -29,16 +29,20 @@ def main(args):
         'associativity': None
     }
 
-    if (args.associativity.startswith("boost")):
-        config['associativity'] = "boost"
-        conf = args.associativity.split(":")
-        if len(conf) > 1: config['boost_power'] = float(conf[1])
-        if len(conf) > 2: config['boost_decay'] = float(conf[2])
-    if (args.associativity.startswith("bytetrack")):
-        config['associativity'] = "bytetrack"
-        conf = args.associativity.split(":")
-        if len(conf) > 1: config['low_conf_threshold'] = float(conf[1])
-        if len(conf) > 2: config['high_conf_threshold'] = float(conf[2])
+    if args.associativity != "":
+        if (args.associativity.startswith("boost")):
+            config['associativity'] = "boost"
+            conf = args.associativity.split(":")
+            if len(conf) > 1: config['boost_power'] = float(conf[1])
+            if len(conf) > 2: config['boost_decay'] = float(conf[2])
+        elif (args.associativity.startswith("bytetrack")):
+            config['associativity'] = "bytetrack"
+            conf = args.associativity.split(":")
+            if len(conf) > 1: config['low_conf_threshold'] = float(conf[1])
+            if len(conf) > 2: config['high_conf_threshold'] = float(conf[2])
+        else:
+            print("INVALID ASSOCIATIVITY TYPE:", args.associativity)
+            return
 
     print("verbose", args.verbose)