changed dataset size

ManifoldRG · May 26, 2021 · 08f2c7f · 08f2c7f
1 parent a0e7bca
commit 08f2c7f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/utils/data.py b/utils/data.py
@@ -25,14 +25,16 @@ def load_dataset(configs):
 def load_cifar10(configs):
  # transform for the training data
  train_transforms = transforms.Compose([
- transforms.RandomCrop(32, padding=4),
+ transforms.Resize(224),
+ # transforms.RandomCrop(32, padding=4),
  transforms.RandomHorizontalFlip(),
  transforms.ToTensor(),
  transforms.Normalize(mean=[0.485, 0.456, 0.406],
  std=[0.229, 0.224, 0.225]),
  ])
 
  test_transforms = transforms.Compose([
+ transforms.Resize(224),
  transforms.ToTensor(),
  transforms.Normalize(mean=[0.485, 0.456, 0.406],
  std=[0.229, 0.224, 0.225]),
@@ -59,14 +61,16 @@ def load_cifar10(configs):
 def load_cifar100(configs):
  # transform for the training data
  train_transforms = transforms.Compose([
- transforms.RandomCrop(32, padding=4),
+ transforms.Resize(224),
+ # transforms.RandomCrop(32, padding=4),
  transforms.RandomHorizontalFlip(),
  transforms.ToTensor(),
  transforms.Normalize(mean=[0.485, 0.456, 0.406],
  std=[0.229, 0.224, 0.225]),
  ])
 
  test_transforms = transforms.Compose([
+ transforms.Resize(224),
  transforms.ToTensor(),
  transforms.Normalize(mean=[0.485, 0.456, 0.406],
  std=[0.229, 0.224, 0.225]),
@@ -94,14 +98,16 @@ def load_cifar100(configs):
 def load_svhn(configs):
  # transform for the training data
  train_transforms = transforms.Compose([
- transforms.RandomCrop(32),
+ transforms.Resize(224),
+ # transforms.RandomCrop(32),
  transforms.RandomHorizontalFlip(),
  transforms.ToTensor(),
  transforms.Normalize(mean=[0.485, 0.456, 0.406],
  std=[0.229, 0.224, 0.225]),
  ])
 
  test_transforms = transforms.Compose([
+ transforms.Resize(224),
  transforms.ToTensor(),
  transforms.Normalize(mean=[0.485, 0.456, 0.406],
  std=[0.229, 0.224, 0.225]),