Create transforms.py

transforms.py

@@ -0,0 +1,12 @@
+from transformers import AutoImageProcessor
+import torchvision.transforms as T
+
+processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
+augment = T.Compose([
+    T.RandomResizedCrop(224),
+    T.RandomHorizontalFlip(),
+    T.ColorJitter(0.2,0.2,0.2,0.1),
+    T.RandomRotation(20),
+    T.ToTensor(),
+    T.Normalize(mean=processor.image_mean, std=processor.image_std)
+])