pytorch torchvision.transforms.RandomRotation填充参数不起作用

b5buobof  于 2023-11-19  发布在  其他
关注(0)|答案(2)|浏览(95)

我正在处理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,),这似乎是一种解决方案,但它对我不起作用。你知道可能是什么问题吗?

wnvonmuf

wnvonmuf1#

您可能使用的是Torchvision v0.8.2或更早版本。此问题已在5个月前的PR #2904中得到修复。如果您未使用v0.9.0或更高版本,则无法在Tensor输入中使用fill:(
所以,唯一的解决办法就是升级你的手电筒。

qzwqbdag

qzwqbdag2#

较新的 Torch 版本的属性名称为'填充',而在2.0.0之前,它是'fillcolor'

相关问题