Crop should be 224, not 96

microsoft · Mar 27, 2023 · 41a10d2 · 41a10d2
1 parent c404e3e
commit 41a10d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/trainers/simclr.py b/torchgeo/trainers/simclr.py
@@ -95,7 +95,7 @@ def __init__(
         # Data augmentation
         # https://github.com/google-research/simclr/blob/master/data_util.py
         self.aug = K.AugmentationSequential(
-            K.RandomResizedCrop(size=(96, 96)),
+            K.RandomResizedCrop(size=(224, 224)),
             K.RandomHorizontalFlip(),
             K.RandomVerticalFlip(),  # added
             # Not appropriate for multispectral imagery, seasonal contrast used instead