oskarastrom commited on
Commit
73afcbb
·
1 Parent(s): 66f8a6f

updated hyperparams

Browse files
scripts/{inferEval.py → infer_eval.py} RENAMED
@@ -29,7 +29,7 @@ def main(args):
29
 
30
  infer(infer_args, config=config, verbose=False)
31
 
32
- evaluate("../frames/result_testing", "../frames/MOT", "../frames/metadata", "tracker", False)
33
 
34
 
35
  def argument_parser():
 
29
 
30
  infer(infer_args, config=config, verbose=False)
31
 
32
+ evaluate("../frames/result", "../frames/MOT", "../frames/metadata", "tracker", False)
33
 
34
 
35
  def argument_parser():
scripts/track_detection.py CHANGED
@@ -28,8 +28,8 @@ def main(args, config={}, verbose=True):
28
  """
29
 
30
  # setup config
31
- if "conf_threshold" not in config: config['conf_threshold'] = 0.001
32
- if "nms_iou" not in config: config['nms_iou'] = 0.6
33
  if "min_length" not in config: config['min_length'] = 0.3
34
  if "max_age" not in config: config['max_age'] = 20
35
  if "iou_threshold" not in config: config['iou_threshold'] = 0.01
@@ -101,6 +101,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
101
  outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
102
 
103
  if config['use_associative']:
 
104
  do_confidence_boost(inference, outputs, verbose=verbose)
105
 
106
  outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
 
28
  """
29
 
30
  # setup config
31
+ if "conf_threshold" not in config: config['conf_threshold'] = 0.3#0.001
32
+ if "nms_iou" not in config: config['nms_iou'] = 0.3#0.6
33
  if "min_length" not in config: config['min_length'] = 0.3
34
  if "max_age" not in config: config['max_age'] = 20
35
  if "iou_threshold" not in config: config['iou_threshold'] = 0.01
 
101
  outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)
102
 
103
  if config['use_associative']:
104
+
105
  do_confidence_boost(inference, outputs, verbose=verbose)
106
 
107
  outputs = do_suppression(inference, conf_thres=config['conf_threshold'], iou_thres=config['nms_iou'], verbose=verbose)