matteopilotto
/

vit-base-patch16-224-in21k-snacks

Image Classification

Inference Endpoints

Model card Files Files and versions Community

matteopilotto commited on May 14, 2022

Commit

a15635e

•

1 Parent(s): 505495c

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ from torchvision.transforms import (
 checkpoint = 'google/vit-base-patch16-224-in21k'
 feature_extractor = ViTFeatureExtractor.from_pretrained(checkpoint)
-# train
 train_aug_transforms = Compose([
     RandomResizedCrop(size=feature_extractor.size),
     RandomHorizontalFlip(p=0.5),
@@ -38,7 +38,7 @@ train_aug_transforms = Compose([
     Normalize(mean=feature_extractor.image_mean, std=feature_extractor.image_std),
 ])
-# validation/test
 valid_aug_transforms = Compose([
     Resize(size=(feature_extractor.size, feature_extractor.size)),
     ToTensor(),

 checkpoint = 'google/vit-base-patch16-224-in21k'
 feature_extractor = ViTFeatureExtractor.from_pretrained(checkpoint)
+# transformations on the training set
 train_aug_transforms = Compose([
     RandomResizedCrop(size=feature_extractor.size),
     RandomHorizontalFlip(p=0.5),
     Normalize(mean=feature_extractor.image_mean, std=feature_extractor.image_std),
 ])
+# transformations on the validation/test set
 valid_aug_transforms = Compose([
     Resize(size=(feature_extractor.size, feature_extractor.size)),
     ToTensor(),