eval: Fix index limits for eval

ethz-asl · Oct 3, 2024 · 9975c22 · 9975c22
1 parent d15f13e
commit 9975c22
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/scripts/evaluation/config.json b/scripts/evaluation/config.json
@@ -4,7 +4,11 @@
         "Agricultural Field": "outdoor0",
         "Rhône Glacier": "rhone_flight"
     },
-    "index": [[0, -1], [0, -1], [1205, 1505]],
+    "index": {
+        "maschinenhalle0": [0, -1],
+        "outdoor0": [0, -1],
+        "rhone_flight": [1205, 1505]
+    },
     "networks": {
         "Metric Depth \\cite{depthanythingv2}-S": "rgb_s_bs8_e9.pth",
         "Metric Depth \\cite{depthanythingv2}-B": "rgb_b_bs4_e8.pth",

diff --git a/scripts/evaluation/evaluate.py b/scripts/evaluation/evaluate.py
@@ -57,7 +57,7 @@ def setup_interface(self, config, scenario_key, network_key):
         self.interface.set_criterion()
 
         dataset_list = [config['scenarios'][scenario_key]]
-        index_list = config.get('index', None)
+        index_list = [config['index'][config["scenarios"][scenario_key]]]
         self.loader, _ = self.interface.get_dataset_loader('val_all', self.datasets_dir, dataset_list, index_list)
 
     def get_results_per_sample(self):