oskarastrom commited on
Commit
14cb5f0
·
1 Parent(s): c6597f3

Update inference.py

Browse files
Files changed (1) hide show
  1. 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=iou_thres, 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=iou_thres, gp=gp, verbose=verbose)
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