CODEWITHSUNDEEP
pythonpytorchdata-augmentationtorchvision
user930958
user930958

Reputation: 3

torchvision.transforms.RandomRotation fill argument not working

I am working on tensors and want to rotate them with torchvision.transforms.RandomRotation and use the fill option.

import torch
import torchvision

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

However, I always get:

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

and it gets not filled with ones. I have the same problem with torchvision.transforms.RandomPerspective. I am using Python 3.8 and PyTorch 1.7.1. I tried using fill=(1,), which seemed to be a workaround, but it did not worked for me. Do you know what could be wrong?

Upvotes: 0

Views: 1967

Answers (2)

R.K
R.K

Reputation: 1839

Newer torch versions have the attribute name as 'fill' while prior to 2.0.0 it was 'fillcolor'

Upvotes: 0

Berriel
Berriel

Reputation: 13641

You're probably using torchvision v0.8.2 or older. This issue was fixed 5 months ago in the PR #2904. If you are not using v0.9.0 or newer, you won't be able to use fill in a Tensor input :(

So, the only solution is to upgrade your torchvision.

Upvotes: 1

Related Questions