associative parameters

inference.py

@@ -330,16 +330,11 @@ def do_tracking(all_preds, image_meter_width, image_meter_height, gp=None, max_a
 
     return json_data
 
-def do_associative_tracking(inference, image_shapes, width, height, image_meter_width, image_meter_height, gp=None, low_thresh=0.2, high_threshold=0.2, max_age=MAX_AGE, iou_thres=IOU_THRES, min_hits=MIN_HITS, min_length=MIN_LENGTH, batch_size=BATCH_SIZE, verbose=True):
+def do_associative_tracking(low_preds, high_preds, image_meter_width, image_meter_height, gp=None, low_thresh=0.2, high_threshold=0.2, max_age=MAX_AGE, iou_thres=IOU_THRES, min_hits=MIN_HITS, min_length=MIN_LENGTH, batch_size=BATCH_SIZE, verbose=True):
 
     if (gp): gp(0, "Tracking...")
 
 
-    low_outputs = do_suppression(inference, conf_thres=low_thresh, iou_thres=0.25, gp=gp, verbose=verbose)
-    low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
-
-    high_outputs = do_suppression(inference, conf_thres=high_threshold, iou_thres=0.25, gp=gp, verbose=verbose)
-    high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
 
 
     # Initialize tracker

scripts/track_detection.py

@@ -101,8 +101,15 @@ 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:
-        results, real_width, real_height = do_associative_tracking(inference, image_shapes, width, height, image_meter_width, image_meter_height, min_length=config['min_length'], max_age=config['max_age'], iou_thres=config['iou_threshold'], min_hits=config['min_hits'], verbose=verbose)
+    if True:
+
+        low_outputs = do_suppression(inference, conf_thres=config['low_conf_threshold'], iou_thres=config['nms_iou'], gp=gp, verbose=verbose)
+        low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
+
+        high_outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], gp=gp, verbose=verbose)
+        high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
+
+        results, real_width, real_height = do_associative_tracking(low_preds, high_preds, image_meter_width, image_meter_height, min_length=config['min_length'], max_age=config['max_age'], iou_thres=config['iou_threshold'], min_hits=config['min_hits'], verbose=verbose)
     else:
         outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 

scripts/track_eval.py

@@ -20,6 +20,8 @@ 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),
@@ -41,6 +43,8 @@ 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.")