add split for supervised evaluation

Orion-AI-Lab · Mar 15, 2023 · 33a27f8 · 33a27f8
1 parent 695d811
commit 33a27f8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/Dataset.py b/Dataset.py
@@ -89,9 +89,9 @@ def __init__(self, config, mode):
         self.negatives = []
         self.frameIDs = []
         self.augmentations = augmentations.get_augmentations(config)
-
+        root_path = self.config['cls_path']
         for file in self.valid_files:
-            annotation = json.load(open(file,'r'))
+            annotation = json.load(open(root_path + file,'r'))
             sample = file.split('/')[-1]
             insar_path = self.config['supervised_data_path'] + '/labeled/'+str(annotation['Deformation'][0])+'/' + sample[:-5] + '.png'
             mask_path = self.config['supervised_data_path'] + '/masks/'+str(annotation['Deformation'][0])+'/' + sample[:-5] + '.png'

diff --git a/configs/supervised_configs.json b/configs/supervised_configs.json
@@ -9,6 +9,7 @@
     "contrastive":false,
     "split_path":"./train_test_split.json",
     "supervised_data_path":"/mnt/nvme1/nbountos/datasets/HephaestusSSL",
+    "cls_path":"/mnt/nvme1/nbountos/datasets/HephaestusSSL/cls_labels/",
     "data_path":"/mnt/nvme1/nbountos/datasets/Hephaestus_Raw",
     "batch_size":64,
     "num_workers":12,

diff --git a/train_test_split.json b/train_test_split.json