Minor update to cryonuseg dataset (#419)

constantinpape · Nov 20, 2024 · b18b9bf · b18b9bf
1 parent 8850d5e
commit b18b9bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/datasets/histopathology/check_cryonuseg.py b/scripts/datasets/histopathology/check_cryonuseg.py
@@ -15,7 +15,7 @@ def check_cryonuseg():
         path=os.path.join(ROOT, "cryonuseg"),
         patch_shape=(1, 512, 512),
         batch_size=1,
-        rater="b3",
+        rater="b1",
         download=True,
     )
     check_loader(loader, 8, rgb=True, instance_labels=True)

diff --git a/torch_em/data/datasets/histopathology/cryonuseg.py b/torch_em/data/datasets/histopathology/cryonuseg.py
@@ -62,7 +62,7 @@ def get_cryonuseg_paths(
         raise ValueError(f"'{rater_choice}' is not a valid rater choice.")
 
     # Point to the instance labels folder
-    label_dir += r"label masks"
+    label_dir += r"label masks modify"
 
     label_paths = natsorted(glob(os.path.join(path, label_dir, "*.tif")))
     raw_paths = natsorted(glob(os.path.join(path, r"tissue images", "*.tif")))