[Error] UserWarning: Argument fill/fillcolor is not supported for Tensor input 에러 해결

 Pytorch를 사용하여 모델을 돌리던 중, augmentation으로써 RandomRotation을 사용하고,

이미지가 돌아갔을 때(rotate) 생기는 여백의 공간을 0으로 채우기 위하여 아래와 같이 코드를 작성하였다.

from torchvision import transforms as transforms

trans = transforms.RandomRotation(degrees=args.aug_degree, fill = (0,))
img_trans = trans(img)

그러나 다음과 같은 에러를 맞이하였다.

UserWarning: Argument fill/fillcolor is not supported for Tensor input

 

알아본 결과 해결방법은 아래와 같다.

 

Torchvision version > v0.9.0으로 설치하기.

conda list를 통해 자신의 torchvision 버전을 확인하고,

만약 버전이 v0.9.0보다 작다면 아래와 같은 방식으로 uninstall 후, v0.9.0으로 설치해주면 된다.

conda uninstall torchvision
pip install torchvision==0.9.0