Fix scale to (1.0, 1.0) to avoid cropping images at all

This fixes the segmentation now. And the results look as expected!
constantinpape · Jan 20, 2025 · f69eb16 · f69eb16
1 parent f66c6d0
commit f69eb16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_em/model/vit.py b/torch_em/model/vit.py
@@ -300,7 +300,7 @@ def transform_inputs(self, x):
             rescale_transform=K.RandomResizedCrop(
                 (448, 448),
                 ratio=(1.0, 1.0),
-                scale=(0.2, 1.0),
+                scale=(1.0, 1.0),
                 resample=Resample.BICUBIC.name,
             ),
             other_transforms=None,