gaussianblur transform not found in torchvision.transforms

user_9863397 0
Viewed 6014

I have written the following data augmentation pipeline for Pytorch:

transform = transforms.Compose([
            transforms.RandomResizedCrop(224),
            transforms.RandomHorizontalFlip(),
            transforms.GaussianBlur(11, sigma=(0.1, 2.0)),
            transforms.ToTensor(),
            transforms.Normalize(mean, std)
        ])

But running the above code gives me the following error:

AttributeError: module 'torchvision.transforms' has no attribute 'GaussianBlur'

Is GaussianBlur a new feature that has not been included in torchvision yet? Or is it just my torchvision version that is too old?

I found it in the following documentation page: torchvision.transforms

Here are my packages versions:

torch                    1.6.0+cu101
torchvision              0.7.0+cu101
pytorch torchvision
2 Answers

On this documentation page, you can look for features on the different versions of pytorch (change version in the upper left corner). It appears that GaussianBlur did not exist in pytorch 1.6, and was added in 1.7. Your code should work with the latest version.

If interested, here is the relevant merge commit : https://github.com/pytorch/vision/pull/2658

user_109357170

You update your pytorch to version 1.7.

conda update --all should work (if you use anaconda distribution).

user_63341190
Related