Use GridSampler for CoNSeP (#645)

kaiko-ai · Sep 25, 2024 · ba045bd · ba045bd
1 parent e182c72
commit ba045bd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/configs/vision/pathology/offline/segmentation/consep.yaml b/configs/vision/pathology/offline/segmentation/consep.yaml
@@ -108,10 +108,7 @@ data:
           init_args: &PREDICT_DATASET_ARGS
             root: ${oc.env:DATA_ROOT, ./data/consep}
             split: train
-            sampler:
-              class_path: eva.vision.data.wsi.patching.samplers.ForegroundGridSampler
-              init_args:
-                max_samples: 25
+            sampler: eva.vision.data.wsi.patching.samplers.GridSampler
             transforms:
               class_path: eva.vision.data.transforms.common.ResizeAndCrop
               init_args:

diff --git a/configs/vision/pathology/online/segmentation/consep.yaml b/configs/vision/pathology/online/segmentation/consep.yaml
@@ -90,10 +90,7 @@ data:
         init_args: &DATASET_ARGS
           root: ${oc.env:DATA_ROOT, ./data/consep}
           split: train
-          sampler:
-            class_path: eva.vision.data.wsi.patching.samplers.ForegroundGridSampler
-            init_args:
-              max_samples: 25
+          sampler: eva.vision.data.wsi.patching.samplers.GridSampler
           transforms:
             class_path: eva.vision.data.transforms.common.ResizeAndCrop
             init_args:

diff --git a/src/eva/vision/data/datasets/segmentation/consep.py b/src/eva/vision/data/datasets/segmentation/consep.py
@@ -37,8 +37,8 @@ def __init__(
         root: str,
         sampler: samplers.Sampler | None = None,
         split: Literal["train", "val"] | None = None,
-        width: int = 224,
-        height: int = 224,
+        width: int = 250,
+        height: int = 250,
         target_mpp: float = 0.25,
         transforms: Callable | None = None,
     ) -> None: