From 3dac470dbee16f34091dd211f4f686f4ca1c9f79 Mon Sep 17 00:00:00 2001
From: Illia Oleksiienko <io@ece.au.dk>
Date: Mon, 12 Dec 2022 22:53:58 +0000
Subject: [PATCH] Fix models
---
run/eval_tracking_3d.py | 35 +++++++++++++++++++++--------------
1 file changed, 21 insertions(+), 14 deletions(-)
diff --git a/run/eval_tracking_3d.py b/run/eval_tracking_3d.py
index 795eb282ca..a36f8e6c3d 100644
--- a/run/eval_tracking_3d.py
+++ b/run/eval_tracking_3d.py
@@ -395,32 +395,39 @@ def test_vnn_tapp(
)
-def all_save_detection_inference(start=0):
+def all_save_detection_inference(start=0, index=0, devices=1):
models = [
# add models
- ("tanet_car", test_tanet),
- ("tapp_car", test_tapp),
- ("pointpillars_car", test_pointpillars),
+ # ("tanet_car", test_tanet),
+ # ("tapp_car", test_tapp),
+ # # ("pointpillars_car", test_pointpillars),
*[("vnn_tanet_car_s" + str(i), test_vnn_tanet) for i in range(1, 3 + 1)],
+ *[("badvnn_tanet_car_s" + str(i), test_vnn_tanet) for i in range(1, 3 + 1)],
*[("vnn_tapp_car_s" + str(i), test_vnn_tapp) for i in range(1, 3 + 1)],
- *[
- ("vnn_pointpillars_car_s" + str(i), test_vnn_pointpillars)
- for i in range(1, 7 + 1)
- ],
- *[
- ("vnna_pointpillars_car_s" + str(i), test_vnna_pointpillars)
- for i in range(1, 4 + 1)
- ],
+ # *[
+ # ("vnn_pointpillars_car_s" + str(i), test_vnn_pointpillars)
+ # for i in range(1, 7 + 1)
+ # ],
+ # *[
+ # ("vnna_pointpillars_car_s" + str(i), test_vnna_pointpillars)
+ # for i in range(1, 4 + 1)
+ # ],
]
- for i, (model_name, test_func) in enumerate(models):
- print(":::", i + 1, "/", len(models))
+ i = index
+
+ while i < len(models):
+
+ print(":::", i, "/", len(models))
if i < start:
+ i += devices
continue
+ model_name, test_func = models[i]
test_func(name=model_name, mode="save_detection_inference")
+ i += devices
if __name__ == "__main__":
fire.Fire()
--
GitLab