oskarastrom commited on
Commit
6697470
1 Parent(s): 2028aea
Files changed (2) hide show
  1. inference.py +1 -1
  2. scripts/track_detection.py +1 -1
inference.py CHANGED
@@ -361,7 +361,7 @@ def do_associative_tracking(low_preds, high_preds, image_meter_width, image_mete
361
 
362
  json_data = tracker.finalize(min_length=min_length)
363
 
364
- return json_data, real_width, real_height
365
 
366
  @patch('json.encoder.c_make_encoder', None)
367
  def json_dump_round_float(some_object, out_path, num_digits=4):
 
361
 
362
  json_data = tracker.finalize(min_length=min_length)
363
 
364
+ return json_data
365
 
366
  @patch('json.encoder.c_make_encoder', None)
367
  def json_dump_round_float(some_object, out_path, num_digits=4):
scripts/track_detection.py CHANGED
@@ -109,7 +109,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
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)
113
  else:
114
  outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
115
 
 
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 = 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)
113
  else:
114
  outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
115