Spaces:
Runtime error
Runtime error
oskarastrom
commited on
Commit
•
2028aea
1
Parent(s):
2316e0f
Update track_detection.py
Browse files
scripts/track_detection.py
CHANGED
@@ -103,10 +103,10 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
|
|
103 |
|
104 |
if True:
|
105 |
|
106 |
-
low_outputs = do_suppression(inference, conf_thres=config['low_conf_threshold'], iou_thres=config['nms_iou'],
|
107 |
low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
|
108 |
|
109 |
-
high_outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'],
|
110 |
high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
|
111 |
|
112 |
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)
|
|
|
103 |
|
104 |
if True:
|
105 |
|
106 |
+
low_outputs = do_suppression(inference, conf_thres=config['low_conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
107 |
low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
|
108 |
|
109 |
+
high_outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
110 |
high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
|
111 |
|
112 |
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)
|