Spaces:
Runtime error
Runtime error
Commit
•
6697470
1
Parent(s):
2028aea
Fix
Browse files- inference.py +1 -1
- scripts/track_detection.py +1 -1
inference.py
CHANGED
@@ -361,7 +361,7 @@ def do_associative_tracking(low_preds, high_preds, image_meter_width, image_mete
|
|
361 |
|
362 |
json_data = tracker.finalize(min_length=min_length)
|
363 |
|
364 |
-
return json_data
|
365 |
|
366 |
@patch('json.encoder.c_make_encoder', None)
|
367 |
def json_dump_round_float(some_object, out_path, num_digits=4):
|
|
|
361 |
|
362 |
json_data = tracker.finalize(min_length=min_length)
|
363 |
|
364 |
+
return json_data
|
365 |
|
366 |
@patch('json.encoder.c_make_encoder', None)
|
367 |
def json_dump_round_float(some_object, out_path, num_digits=4):
|
scripts/track_detection.py
CHANGED
@@ -109,7 +109,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
|
|
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
|
113 |
else:
|
114 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
115 |
|
|
|
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 = 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)
|
113 |
else:
|
114 |
outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
|
115 |
|