def _get_cutout_train_transform(self):
transform = torchvision.transforms.Compose([
transforms.normalize(self.mean, self.std),
transforms.cutout(self.config['cutout_size'],
self.config['cutout_prob'],
self.config['cutout_inside']),
transforms.to_tensor(),
])
return transform
def _get_cutout_train_transform(self):
transform = torchvision.transforms.Compose([
torchvision.transforms.RandomCrop(28, padding=4),
torchvision.transforms.RandomHorizontalFlip(),
transforms.normalize(self.mean, self.std),
transforms.cutout(self.config['cutout_size'],
self.config['cutout_prob'],
self.config['cutout_inside']),
transforms.to_tensor(),
])
return transform
def _add_cutout(self):
transform = transforms.cutout(self.config['cutout_size'],
self.config['cutout_prob'],
self.config['cutout_inside'])
self._train_transforms.append(transform)