在 torchvision.transforms 中找不到高斯模糊变换

Question

我为 Pytorch 编写了以下数据增强管道：

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

但是运行上面的代码给我以下错误：

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

GaussianBlur 是 torchvision 还没有加入的新功能吗？还是我的 torchvision 版本太旧了？

我在以下文档页面中找到了它：torchvision.transforms

这是我的软件包版本：

torch                    1.6.0+cu101
torchvision              0.7.0+cu101

Answer 1

在this documentation page，你可以在不同版本的pytorch上寻找特性（在左上角更改版本）。貌似 GaussianBlur 在 pytorch 1.6 中不存在，是在 1.7 中添加的。您的代码应该适用于最新版本。

如果有兴趣，这里是相关的合并提交：https://github.com/pytorch/vision/pull/2658

Answer 2

您将 pytorch 更新到 1.7 版本。

conda update --all 应该有效（如果您使用 anaconda 发行版）。

在 torchvision.transforms 中找不到高斯模糊变换

gaussianblur transform not found in torchvision.transforms

pytorch

torchvision