torchvision.transforms.RandomRotation填充参数无效

Question

我正在研究张量，希望用torchvision.transforms.RandomRotation旋转它们，并使用fill选项。

import torch
import torchvision

img1 = torch.rand((1, 16, 16))
img2 = torchvision.transforms.RandomRotation(45, fill=1)(img1)

然而，我总是得到：

参数填充/填充颜色不支持张量输入。填充值为零

而且里面没有装满一个。我对torchvision.transforms.RandomPerspective也有同样的问题。我正在使用Python3.8和PyTorch 1.7.1。我试过使用fill=(1,)，这似乎是一种解决办法，但对我来说没有用。你知道会有什么问题吗？

Answer 1

您可能使用的是torchvision v0.8.2或更高版本。这个问题是5个月前在PR 2904上解决的。如果您不使用v0.9.0或更新版本，则无法在Tensor输入中使用fill：

所以，唯一的解决办法就是升级你的火炬视觉。