Update track_eval.py

scripts/track_eval.py

@@ -24,7 +24,8 @@ def main(args):
         'min_length': float(args.min_length),
         'max_age': int(args.max_age),
         'iou_threshold': float(args.iou_threshold),
-        'min_hits': int(args.min_hits)
+        'min_hits': int(args.min_hits),
+        'use_associative': args.use_associative
     }
 
     track(infer_args, config=config, verbose=False)
@@ -34,13 +35,14 @@ def main(args):
 
 def argument_parser():
     parser = argparse.ArgumentParser()
+    parser.add_argument("--detection_dir", default="../frames/detection_storage")
     parser.add_argument("--conf_threshold", default=0.3, help="Config object. Required.")
     parser.add_argument("--nms_iou", default=0.3, help="Config object. Required.")
     parser.add_argument("--min_length", default=0.3, help="Config object. Required.")
     parser.add_argument("--max_age", default=20, help="Config object. Required.")
     parser.add_argument("--iou_threshold", default=0.01, help="Config object. Required.")
     parser.add_argument("--min_hits", default=11, help="Config object. Required.")
-    parser.add_argument("--detection_dir", default="../frames/detection_storage")
+    parser.add_argument("--use_associative", action='store_true', help="Config object. Required.")
     return parser
 
 if __name__ == "__main__":