AssoTracker

inference.py

@@ -366,7 +366,7 @@ def do_associative_tracking(inference, image_shapes, width, height, image_meter_
 
     json_data = tracker.finalize(min_length=min_length)
 
-    return json_data
+    return json_data, real_width, real_height
 
 @patch('json.encoder.c_make_encoder', None)
 def json_dump_round_float(some_object, out_path, num_digits=4):

scripts/track_detection.py

@@ -100,22 +100,20 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
                 image_meter_width = sequence['x_meter_stop'] - sequence['x_meter_start']
                 image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
 
-    print(len(inference))
-    print(len(inference[0]))
-    print(inference[0][0])
 
-    results = 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)
+    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)
 
-    outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
+    if False:
+        outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 
 
-    if config['use_associative']:
+        if config['use_associative']:
 
-        do_confidence_boost(inference, outputs, conf_power=config['boost_power'], conf_decay=config['boost_decay'], verbose=verbose)
+            do_confidence_boost(inference, outputs, conf_power=config['boost_power'], conf_decay=config['boost_decay'], verbose=verbose)
 
-        outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
+            outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 
-    all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height, verbose=verbose)
+        all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height, verbose=verbose)
 
 
     mot_rows = []

scripts/track_eval.py

@@ -30,6 +30,8 @@ def main(args):
         'use_associative': args.use_associative
     }
 
+    print("verbose", args.verbose)
+
     track(infer_args, config=config, verbose=args.verbose)
 
     evaluate(infer_args.output, "../frames/MOT", "../frames/metadata", infer_args.tracker, False)