Spaces:
Runtime error
Runtime error
Commit
·
ae15d0d
1
Parent(s):
4b6e487
Update track_detection.py
Browse files
scripts/track_detection.py
CHANGED
@@ -100,13 +100,13 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
|
|
100 |
image_meter_width = sequence['x_meter_stop'] - sequence['x_meter_start']
|
101 |
image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
|
102 |
|
103 |
-
|
|
|
|
|
104 |
|
105 |
-
|
106 |
-
print(len(outputs[0]))
|
107 |
-
print(outputs[0][0])
|
108 |
|
109 |
-
|
110 |
|
111 |
|
112 |
if config['use_associative']:
|
|
|
100 |
image_meter_width = sequence['x_meter_stop'] - sequence['x_meter_start']
|
101 |
image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
|
102 |
|
103 |
+
print(len(inference))
|
104 |
+
print(len(inference[0]))
|
105 |
+
print(inference[0][0])
|
106 |
|
107 |
+
results = do_associative_tracking(inference, 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 |
|
111 |
|
112 |
if config['use_associative']:
|