我正在处理Tensor,想用torchvision.transforms.RandomRotation
旋转它们,并使用fill
选项。
import torch
import torchvision
img1 = torch.rand((1, 16, 16))
img2 = torchvision.transforms.RandomRotation(45, fill=1)(img1)
字符串
但我总是得到:
Tensor输入不支持参数fill/fillcolor。填充值为零
而且它没有被1填充。我在使用torchvision.transforms.RandomPerspective
时也遇到了同样的问题。我使用的是Python 3.8和PyTorch 1.7.1。我尝试使用fill=(1,)
,这似乎是一种解决方案,但它对我不起作用。你知道可能是什么问题吗?
2条答案
按热度按时间wnvonmuf1#
您可能使用的是Torchvision v0.8.2或更早版本。此问题已在5个月前的PR #2904中得到修复。如果您未使用v0.9.0或更高版本,则无法在
Tensor
输入中使用fill
:(所以,唯一的解决办法就是升级你的手电筒。
qzwqbdag2#
较新的 Torch 版本的属性名称为'填充',而在2.0.0之前,它是'fillcolor'