Merge pull request #1550 from mikel-brostrom/delete-double-ci-mot-gen

fix eval in ci

tracking/evolve.py

@@ -4,11 +4,13 @@
 from pathlib import Path
 
 from boxmot.utils.checks import RequirementsChecker
+from boxmot.utils import EXAMPLES
 from tracking.val import (
     run_generate_dets_embs,
     run_generate_mot_results,
     run_trackeval,
-    parse_opt as parse_optt
+    parse_opt as parse_optt,
+    download_mot_eval_tools
 )
 from boxmot.utils import ROOT, NUM_THREADS
 
@@ -26,6 +28,7 @@ def __init__(self, opt, parameters):
         self.opt = opt
 
     def objective_function(self, config):
+        download_mot_eval_tools(self.opt.val_tools_path)
         # generate new set of mot challenge compliant results with
         # new set of generated tracker parameters
         run_generate_mot_results(self.opt, config)
@@ -130,6 +133,7 @@ def get_search_space(tracking_method):
     return search_space
 
 opt = parse_optt()
+opt.val_tools_path = EXAMPLES / 'val_utils'
 opt.source = Path(opt.source).resolve()
 opt.yolo_model = [Path(y).resolve() for y in opt.yolo_model]
 opt.reid_model = [Path(r).resolve() for r in opt.reid_model]

tracking/val.py

@@ -376,6 +376,7 @@ def parse_opt() -> argparse.Namespace:
     parser.add_argument('--agnostic-nms', default=False, action='store_true', help='class-agnostic NMS')
     parser.add_argument('--n-trials', type=int, default=4, help='nr of trials for evolution')
     parser.add_argument('--objectives', type=str, nargs='+', default=["HOTA", "MOTA", "IDF1"], help='set of objective metrics: HOTA,MOTA,IDF1')
+    parser.add_argument('--val-tools-path', type=Path, default=EXAMPLES / 'val_utils', help='path to store trackeval repo in')
 
     subparsers = parser.add_subparsers(dest='command')
 
@@ -405,7 +406,6 @@ def parse_opt() -> argparse.Namespace:
     opt = parse_opt()
 
     # download MOT benchmark
-    opt.val_tools_path = EXAMPLES / 'val_utils'
     download_mot_eval_tools(opt.val_tools_path)
     zip_path = download_mot_dataset(opt.val_tools_path, opt.benchmark)
     unzip_mot_dataset(zip_path, opt.val_tools_path, opt.benchmark)