updated panda config (#437)

kaiko-ai · May 13, 2024 · 7fb4445 · 7fb4445
1 parent 1211e5a
commit 7fb4445
Showing 1 changed file with 26 additions and 18 deletions.
diff --git a/configs/vision/dino_vit/offline/panda.yaml b/configs/vision/dino_vit/offline/panda.yaml
@@ -97,23 +97,32 @@ data:
           <<: *DATASET_ARGS
           split: test
       predict:
-        class_path: eva.vision.datasets.WsiClassificationDataset
-        init_args:
-          root: ${oc.env:DATA_ROOT, ./slide_data}/panda
-          manifest_file: manifest.csv
-          sampler:
-            class_path: eva.vision.data.wsi.patching.samplers.ForegroundGridSampler
-            init_args:
-              max_samples: 100
-          width: 224
-          height: 224
-          target_mpp: 0.5
-          image_transforms:
-            class_path: eva.vision.data.transforms.common.ResizeAndCrop
-            init_args:
-              size: ${oc.env:RESIZE_DIM, 224} 
-              mean: ${oc.env:NORMALIZE_MEAN, [0.485, 0.456, 0.406]} 
-              std: ${oc.env:NORMALIZE_STD, [0.229, 0.224, 0.225]}
+        - class_path: eva.vision.datasets.WsiClassificationDataset
+          init_args: &PREDICT_DATASET_ARGS
+            root: ${oc.env:DATA_ROOT, ./slide_data}/panda
+            manifest_file: manifest.csv
+            sampler:
+              class_path: eva.vision.data.wsi.patching.samplers.ForegroundGridSampler
+              init_args:
+                max_samples: 100
+            width: 224
+            height: 224
+            target_mpp: 0.5
+            split: train
+            image_transforms:
+              class_path: eva.vision.data.transforms.common.ResizeAndCrop
+              init_args:
+                size: ${oc.env:RESIZE_DIM, 224} 
+                mean: ${oc.env:NORMALIZE_MEAN, [0.485, 0.456, 0.406]} 
+                std: ${oc.env:NORMALIZE_STD, [0.229, 0.224, 0.225]}
+        - class_path: eva.vision.datasets.WsiClassificationDataset
+          init_args:
+            <<: *PREDICT_DATASET_ARGS
+            split: val
+        - class_path: eva.vision.datasets.WsiClassificationDataset
+          init_args:
+            <<: *PREDICT_DATASET_ARGS
+            split: test
     dataloaders:
       train:
         batch_size: &BATCH_SIZE ${oc.env:BATCH_SIZE, 16}
@@ -124,4 +133,3 @@ data:
         batch_size: &PREDICT_BATCH_SIZE ${oc.env:PREDICT_BATCH_SIZE, 64}
         num_workers: 12 #multiprocessing.cpu_count
         prefetch_factor: 2
-