| 545 | return tensor_cropped |
| 546 | |
| 547 | class HorizontalFlip(object): |
| 548 | def __call__(self, img): |
| 549 | return F.hflip(img) |
| 550 | |
| 551 | def get_transform(size=(512,512),apply_agmentationt=False,normalization=False,flip=False): |
| 552 | color_jitter = transforms.ColorJitter( |