associativity types

inference.py

@@ -258,7 +258,6 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, b
     boost_range = math.floor(math.sqrt(1/boost_decay * math.log(boost_power / boost_cutoff)))
     boost_scale = boost_power * math.exp(-boost_decay)
     
-    outputs = []
     with tqdm(total=len(inference), desc="Running confidence boost", ncols=0, disable=not verbose) as pbar:
         for batch_i in range(len(inference)):
 
@@ -297,7 +296,13 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, b
 def boost_frame(safe_frame, base_frame, dt, power=1, decay=1):
     safe_boxes = safe_frame[:, :4]
     boxes = xywh2xyxy(base_frame[:, :4])  # center_x, center_y, width, height) to (x1, y1, x2, y2)≈
-    ious = box_iou(boxes.double(), safe_boxes).float()
+
+    # If running on CPU, you have to convert to double for the .prod() function in box_iou for some reason?
+    if torch.cuda.is_available():
+        ious = box_iou(boxes, safe_boxes)
+    else: 
+        ious = box_iou(boxes.double(), safe_boxes).float()
+    
     score = torch.matmul(ious, safe_frame[:, 4])
     # score = iou(safe_box, base_box) * confidence(safe_box)
     base_frame[:, 4] *= 1 + power*(score)*math.exp(-decay*(dt*dt-1))

scripts/track_eval.py

@@ -21,18 +21,25 @@ def main(args):
 
     config = {
         'conf_threshold': float(args.conf_threshold),
-        'low_conf_threshold': float(args.low_conf),
-        'high_conf_threshold': float(args.high_conf),
         'nms_iou': float(args.nms_iou),
         'min_length': float(args.min_length),
         'max_age': int(args.max_age),
         'iou_threshold': float(args.iou_threshold),
         'min_hits': int(args.min_hits),
-        'boost_power': float(args.boost_power),
-        'boost_decay': float(args.boost_decay),
-        'use_associative': args.use_associative
+        '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])
+
     print("verbose", args.verbose)
 
     track(infer_args, config=config, verbose=args.verbose)
@@ -51,16 +58,12 @@ def argument_parser():
     parser = argparse.ArgumentParser()
     parser.add_argument("--detection_dir", default="../frames/detection_storage")
     parser.add_argument("--conf_threshold", default=0.3, help="Config object. Required.")
-    parser.add_argument("--low_conf", default=0.01, help="Config object. Required.")
-    parser.add_argument("--high_conf", default=0.3, help="Config object. Required.")
     parser.add_argument("--nms_iou", default=0.3, help="Config object. Required.")
     parser.add_argument("--min_length", default=0.3, help="Config object. Required.")
     parser.add_argument("--max_age", default=20, help="Config object. Required.")
     parser.add_argument("--iou_threshold", default=0.01, help="Config object. Required.")
     parser.add_argument("--min_hits", default=11, help="Config object. Required.")
-    parser.add_argument("--boost_power", default=1, help="Config object. Required.")
-    parser.add_argument("--boost_decay", default=1, help="Config object. Required.")
-    parser.add_argument("--use_associative", action='store_true', help="Config object. Required.")
+    parser.add_argument("--associativity", default='', help="Config object. Required.")
     parser.add_argument("--verbose", action='store_true', help="Config object. Required.")
     return parser