Spaces:
Runtime error
Runtime error
Commit
·
14cb5f0
1
Parent(s):
c6597f3
Update inference.py
Browse files- inference.py +2 -2
inference.py
CHANGED
@@ -335,10 +335,10 @@ def do_associative_tracking(inference, image_shapes, width, height, image_meter_
|
|
335 |
if (gp): gp(0, "Tracking...")
|
336 |
|
337 |
|
338 |
-
low_outputs = do_suppression(inference, conf_thres=low_thresh, iou_thres=
|
339 |
low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
|
340 |
|
341 |
-
high_outputs = do_suppression(inference, conf_thres=high_threshold, iou_thres=
|
342 |
high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
|
343 |
|
344 |
|
|
|
335 |
if (gp): gp(0, "Tracking...")
|
336 |
|
337 |
|
338 |
+
low_outputs = do_suppression(inference, conf_thres=low_thresh, iou_thres=0.25, gp=gp, verbose=verbose)
|
339 |
low_preds, real_width, real_height = format_predictions(image_shapes, low_outputs, width, height, verbose=verbose)
|
340 |
|
341 |
+
high_outputs = do_suppression(inference, conf_thres=high_threshold, iou_thres=0.25, gp=gp, verbose=verbose)
|
342 |
high_preds, real_width, real_height = format_predictions(image_shapes, high_outputs, width, height, verbose=verbose)
|
343 |
|
344 |
|