AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 04:00:46 +02:00
Code Issues Projects Releases Packages Wiki Activity

Enleve dependance RandAugment

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-31 10:36:50 -05:00
parent cd6e159b77
commit 250ce2c3cf
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
higher/smart_aug/datasets.py
View file

@ -30,13 +30,13 @@ transform = torchvision.transforms.Compose([
# torchvision.transforms.Normalize((0.4914, 0.4822, 0.4465), (0.2023, 0.1994, 0.2010)), #CIFAR10 # torchvision.transforms.Normalize((0.4914, 0.4822, 0.4465), (0.2023, 0.1994, 0.2010)), #CIFAR10
]) ])
from RandAugment import RandAugment
# Add RandAugment with N, M(hyperparameter)
transform_train = torchvision.transforms.Compose([ transform_train = torchvision.transforms.Compose([
#transforms.RandomHorizontalFlip(), #transforms.RandomHorizontalFlip(),
#transforms.RandomVerticalFlip(), #transforms.RandomVerticalFlip(),
torchvision.transforms.ToTensor(), torchvision.transforms.ToTensor(),
]) ])
#from RandAugment import RandAugment
# Add RandAugment with N, M(hyperparameter)
#transform_train.transforms.insert(0, RandAugment(n=2, m=30)) #transform_train.transforms.insert(0, RandAugment(n=2, m=30))
### Classic Dataset ### ### Classic Dataset ###