[Error] Pytorch - transforms.RandomRotation : function takes exactly 1 argument (3 given) 에러 해결

Pytorch-torchvision의 transforms 라이브러리를 사용하여 Data augmentation을 진행하던 도중 마주친 에러.

TypeError: function takes exactly 1 argument (3 given)

Stack overflow에 따르면 torch와 Pillow version의 호환문제로 인해 발생한 것으로,

다음과 같이  fill = (0,)  을 RandomRotation 함수에 추가하여 해결할 수 있다.

 

aug_methods = transforms.RandomRotation(30, expand=False, fill=(0,))

 

아래는 코드 전문.