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)

然而，我总是得到：

Argument fill/fillcolor is not supported for Tensor input. Fill value is zero

而且里面没有填满。我对 torchvision.transforms.RandomPerspective 也有同样的问题。 我正在使用 Python 3.8 和 PyTorch 1.7.1。我尝试使用 fill=(1,)，这似乎是一种解决方法，但它对我不起作用。你知道哪里出了问题吗？

Answer 1

您可能使用的是 torchvision v0.8.2 或更早版本。此问题已在 5 个月前在 PR #2904. If you are not using v0.9.0 或更新版本中修复，您将无法在 Tensor 输入中使用 fill :(

所以，唯一的解决办法就是升级你的 torchvision。