(mean, std, crop_size, train=True)
| 178 | |
| 179 | |
| 180 | def get_transform(mean, std, crop_size, train=True): |
| 181 | if train: |
| 182 | return transforms.Compose([transforms.RandomHorizontalFlip(), |
| 183 | transforms.RandomCrop(crop_size, padding=4, padding_mode='reflect'), |
| 184 | transforms.ToTensor(), |
| 185 | transforms.Normalize(mean, std)]) |
| 186 | else: |
| 187 | return transforms.Compose([transforms.ToTensor(), |
| 188 | transforms.Normalize(mean, std)]) |
| 189 | |
| 190 | |
| 191 | class SSL_Dataset: |