Fix issue that slows down preprocessing

FR-DC · Jun 11, 2024 · aa5292e · aa5292e
1 parent 8e17245
commit aa5292e
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/model_tests/utils.py b/tests/model_tests/utils.py
@@ -14,6 +14,7 @@
     RandomRotation,
     RandomApply,
     Resize,
+    RandomResizedCrop,
 )
 from torchvision.transforms.v2 import RandomHorizontalFlip
 
@@ -53,11 +54,12 @@ def rand_weak_aug(size: int):
         [
             ToImage(),
             ToDtype(torch.float32, scale=True),
+            RandomResizedCrop(
+                size, scale=(0.08, 1.0), ratio=(0.95, 1.05), antialias=True
+            ),
             RandomHorizontalFlip(),
             RandomVerticalFlip(),
             RandomApply([RandomRotation((90, 90))], p=0.5),
-            Resize(size, antialias=True),
-            CenterCrop(size),
         ]
     )
 
@@ -78,10 +80,11 @@ def rand_strong_aug(size: int):
         [
             ToImage(),
             ToDtype(torch.float32, scale=True),
+            RandomResizedCrop(
+                size, scale=(0.08, 1.0), ratio=(0.9, 1.1), antialias=True
+            ),
             RandomHorizontalFlip(),
             RandomVerticalFlip(),
             RandomApply([RandomRotation((90, 90))], p=0.5),
-            Resize(size, antialias=True),
-            RandomCrop(size, pad_if_needed=False),  # Strong
         ]
     )