python - 使用 torchvision 下载 celebA 数据集时出错

Question

使用 torchvision 模块数据集，我无法下载 celebA 图像数据集。我很确定我做的一切都是正确的。

dataset = datasets.CelebA(
    root='../datasets/celebA/train_images',
    split='train',
    target_type='identity',
    transform=transforms.Compose([transforms.ToTensor()]),
    download=True)

错误:

BadZipFile: File is not a zip file

Answer 1

这是一个已知问题，已在 #1920 中报告过, 它似乎已在 #4109 中修复但该提交尚未包含在稳定版本中。

与此同时，您可以执行以下操作:

• 查看datasets.CelebA的源代码，搜索 file_list 并记下其中的文件。
• 从 here 下载这些文件，并将文件复制到名为 celeba
的新目录中
• 将img_align_celeba.zip解压到同一目录

最后使用 download=False 调用 datasets.CelebA()。