oskarastrom commited on
Commit
ae15d0d
·
1 Parent(s): 4b6e487

Update track_detection.py

Browse files
Files changed (1) hide show
  1. scripts/track_detection.py +5 -5
scripts/track_detection.py CHANGED
@@ -100,13 +100,13 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
100
  image_meter_width = sequence['x_meter_stop'] - sequence['x_meter_start']
101
  image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
102
 
103
- outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 
 
104
 
105
- print(len(outputs))
106
- print(len(outputs[0]))
107
- print(outputs[0][0])
108
 
109
- results = do_associative_tracking(outputs, 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)
110
 
111
 
112
  if config['use_associative']:
 
100
  image_meter_width = sequence['x_meter_stop'] - sequence['x_meter_start']
101
  image_meter_height = sequence['y_meter_start'] - sequence['y_meter_stop']
102
 
103
+ print(len(inference))
104
+ print(len(inference[0]))
105
+ print(inference[0][0])
106
 
107
+ results = do_associative_tracking(inference, 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
 
111
 
112
  if config['use_associative']: