Spaces:
Runtime error
Runtime error
| from torchvision import transforms | |
| from PIL import Image | |
| import argparse | |
| def main(): | |
| args = setup_parser().parse_args() | |
| path = args.path | |
| img = Image.open(path) | |
| trf = transforms.Compose([ | |
| transforms.RandomAffine(25, translate=(0.1, 0.1), scale=(0.9, 1.1), shear=8), | |
| transforms.RandomResizedCrop(224), | |
| transforms.RandomHorizontalFlip(), | |
| transforms.ColorJitter(brightness = 0.3, saturation = 0.2), | |
| transforms.RandomApply([transforms.GaussianBlur(kernel_size=5, sigma=(0.5, 2.0))], p=1), # Apply Gaussian blur with random probability | |
| ]) | |
| img = trf(img) | |
| img.save("blur.jpg") | |
| def setup_parser(): | |
| parser = argparse.ArgumentParser(description='Reproduce of multiple continual learning algorthms.') | |
| parser.add_argument('--path', type=str, | |
| help='Image file.') | |
| return parser | |
| if __name__ == '__main__': | |
| main() | |