From 33f247a4de8afa8790ac3d7e197cf75b5814c069 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mikel=20Brostr=C3=B6m?= Date: Sun, 28 Jul 2024 23:49:24 +0200 Subject: [PATCH 1/4] fix --- tracking/val.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tracking/val.py b/tracking/val.py index 5cf20dcb05..0984174764 100644 --- a/tracking/val.py +++ b/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) From da82f72532624bc5a99e12b628e0980859eae676 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mikel=20Brostr=C3=B6m?= Date: Sun, 28 Jul 2024 23:53:41 +0200 Subject: [PATCH 2/4] fix --- tracking/evolve.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tracking/evolve.py b/tracking/evolve.py index 4253eb6e95..e4cb17b1ba 100644 --- a/tracking/evolve.py +++ b/tracking/evolve.py @@ -130,6 +130,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] From ea5e1d9b215cb17b07be5001822275b923b438ed Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mikel=20Brostr=C3=B6m?= Date: Sun, 28 Jul 2024 23:55:44 +0200 Subject: [PATCH 3/4] fix --- tracking/evolve.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tracking/evolve.py b/tracking/evolve.py index e4cb17b1ba..11f4aec1e5 100644 --- a/tracking/evolve.py +++ b/tracking/evolve.py @@ -4,6 +4,7 @@ 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, From 28940af96621e3785795e70cb63c4c9be2edf495 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mikel=20Brostr=C3=B6m?= Date: Mon, 29 Jul 2024 00:03:50 +0200 Subject: [PATCH 4/4] downlaod trackeval --- tracking/evolve.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tracking/evolve.py b/tracking/evolve.py index 11f4aec1e5..1f423451fc 100644 --- a/tracking/evolve.py +++ b/tracking/evolve.py @@ -9,7 +9,8 @@ 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 @@ -27,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)