oskarastrom commited on
Commit
c0dc29f
·
1 Parent(s): b9c5fdf

Update track_eval.py

Browse files
Files changed (1) hide show
  1. scripts/track_eval.py +2 -1
scripts/track_eval.py CHANGED
@@ -13,7 +13,7 @@ class Object(object):
13
  def main(args):
14
 
15
  infer_args = Object()
16
- infer_args.detections = "../frames/detection_storage"
17
  infer_args.metadata = "../frames/metadata"
18
  infer_args.output = "../frames/result_testing"
19
  infer_args.tracker = 'tracker'
@@ -40,6 +40,7 @@ def argument_parser():
40
  parser.add_argument("--max_age", default=20, help="Config object. Required.")
41
  parser.add_argument("--iou_threshold", default=0.01, help="Config object. Required.")
42
  parser.add_argument("--min_hits", default=11, help="Config object. Required.")
 
43
  return parser
44
 
45
  if __name__ == "__main__":
 
13
  def main(args):
14
 
15
  infer_args = Object()
16
+ infer_args.detections = args.detection_dir
17
  infer_args.metadata = "../frames/metadata"
18
  infer_args.output = "../frames/result_testing"
19
  infer_args.tracker = 'tracker'
 
40
  parser.add_argument("--max_age", default=20, help="Config object. Required.")
41
  parser.add_argument("--iou_threshold", default=0.01, help="Config object. Required.")
42
  parser.add_argument("--min_hits", default=11, help="Config object. Required.")
43
+ parser.add_argument("--detection_dir", default="../frames/detection_storage")
44
  return parser
45
 
46
  if __name__ == "__main__":