(args)
| 135 | |
| 136 | |
| 137 | def get_default_train_trans(args): |
| 138 | normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], |
| 139 | std=[0.229, 0.224, 0.225]) |
| 140 | if (not hasattr(args, 'resolution')) or args.resolution == 224: |
| 141 | trans = transforms.Compose([ |
| 142 | transforms.RandomResizedCrop(224), |
| 143 | transforms.RandomHorizontalFlip(), |
| 144 | transforms.ToTensor(), |
| 145 | normalize]) |
| 146 | else: |
| 147 | raise ValueError('Not yet implemented.') |
| 148 | return trans |
| 149 | |
| 150 | |
| 151 | def get_default_val_trans(args): |