[연구실] Pillow version 문제

<Pillow version>

다음과 같이 image를 Transform할려고 하니 자꾸 말도 안되는 error가 발생

자꾸, image 하나의 data를 넣었는데 두개의 params를 넣어야한다는 error가 발생

 

#image transform & cuda data로 바꿔주기
    def image_transform(self, img_list):

        # print(img_list) - debug

        # data transform
        transform = transforms.Compose([
        transforms.Resize((224,224)),
        transforms.ToTensor(),
        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
        ])

        # cuda
        cuda = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

        result_image = []

        for image in img_list:
            # print(image)
            img = transform(image).to(cuda)
            result_image.append(img)

        return result_image

 

알고보니 Pillow의 version을 낮추면 바로 해결되었다. 다음과 같이 requirements.txt 파일을 생성하여 version을 낮추고 명시

 

torch==1.9.0

torchaudio==0.9.0

torchvision==0.10.0

Pillow==8.3.1

beautifulsoup4==4.9.3

numpy==1.19.5