oskarastrom commited on
Commit
3986029
·
1 Parent(s): 39652d5

Update track_detection.py

Browse files
Files changed (1) hide show
  1. scripts/track_detection.py +1 -1
scripts/track_detection.py CHANGED
@@ -104,7 +104,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
104
  print(len(inference[0]))
105
  print(inference[0][0])
106
 
107
- results = do_associative_tracking(inference, width, height, 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
 
 
104
  print(len(inference[0]))
105
  print(inference[0][0])
106
 
107
+ results = do_associative_tracking(inference, image_shapes, width, height, 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