oskarastrom commited on
Commit
2028aea
1 Parent(s): 2316e0f

Update track_detection.py

Browse files
Files changed (1) hide show
  1. scripts/track_detection.py +2 -2
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'], gp=gp, 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'], gp=gp, 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)
 
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)