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 |
|