Spaces:
Runtime error
Runtime error
Commit
·
3986029
1
Parent(s):
39652d5
Update track_detection.py
Browse files
scripts/track_detection.py
CHANGED
@@ -104,7 +104,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
|
|
104 |
print(len(inference[0]))
|
105 |
print(inference[0][0])
|
106 |
|
107 |
-
results = do_associative_tracking(inference, 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)
|
108 |
|
109 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
110 |
|
|
|
104 |
print(len(inference[0]))
|
105 |
print(inference[0][0])
|
106 |
|
107 |
+
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)
|
108 |
|
109 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
110 |
|