torchvision.datasets.CelebA 사용시 file is not a zip file 오류

개요

얼굴 데이터셋인 CelebA 데이터셋을 쓰기 위해서 torchvision.datasets.CelebA을 사용할 때 file is not a zip file 오류가 발생하는 경우가 있다.

상황 및 원인

import torchvision.datasets as dset

dataset = dset.CelebA(".", split="train", download=True)

download=True옵션으로 구글 드라이브를 통해서 파일을 다운 받는데 트레픽 때문에 오류가 발생한다.
정해진 path에 가보면 파일이 있긴 한데, 열어보면 제대로 된 파일이 아니다..

솔루션

구글 드러이브를 통해 다운받는 파일은 아래 6개 이다.
각 파일을 열어 보고 제대로된 내용을 확인하지 못했다면 구글 드라이브 트래픽 이슈일 가능성이 있다.

• img_align_celeba.zip
• list_attr_celeba.txt
• identity_CelebA.txt
• list_bbox_celeba.txt
• list_landmarks_align_celeba.txt
• list_eval_partition.txt

구글 드라이브에서 수동으로 파일을 다운받아 해결 할 수 있다.