Merge branch 'f/adjust_path_dresden_capsule' into 'main'

F/adjust path dresden capsule See merge request es/ai/hannah/hannah!391
ekut-es · Jun 10, 2024 · 0029169 · 0029169
2 parents b9965d7 + a6287f3
commit 0029169
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hannah/conf/dataset/dresden_capsule.yaml b/hannah/conf/dataset/dresden_capsule.yaml
@@ -25,11 +25,11 @@ dataset: dresden_capsule
 sampler: random
 weighted_loss: false
 
-task: sections # Check splits folder for other task options.
+task: section # Check splits folder for other task options.
 split: split_0
 
 downsampling:
-  enabled: true
+  enabled: false
   ratio:
     binary: 1 # ratio normal : anomaly
     # Proportions of each class that should be used.

diff --git a/hannah/datasets/vision/dresden_capsule.py b/hannah/datasets/vision/dresden_capsule.py
@@ -92,9 +92,9 @@ def prepare(cls, config):
 
     @classmethod
     def splits(cls, config):
-        data_folder = pathlib.Path(config["data_folder"]) / "dresden-capsule"
+        data_folder = pathlib.Path(config["data_folder"]) / "galar"
         study_folder = data_folder / "images"
-        split_folder = data_folder / "splits_tuebingen" / config.task
+        split_folder = data_folder / "splits_publication" / config.task
 
         test_data = pd.read_csv(split_folder / "test.csv")
         val_data = pd.read_csv(split_folder / config.split / "val.csv")