PyTorch中手动配置CelebA数据集：解决路径引发的RuntimeError

你的问题出在目录结构不符合PyTorch Lightning（底层依赖torchvision的CelebA数据集类）的预期，调整一下就行：

• 正确的目录结构应该是：

  $dataroot/
      img_align_celeba/
          *.jpg  # 所有对齐后的图片
      list_attr_celeba.txt
      list_bbox_celeba.txt
      list_eval_partition.txt
      # 其他从官网下载的元数据文本文件
  

• 你当前的结构是$dataroot/celeba/img_align_celeba/，相当于多套了一层celeba子目录，数据集类找不到对应的文件。

调整步骤：

1. 把$dataroot/celeba/img_align_celeba整个文件夹移到$dataroot根目录下
2. 同时把你下载的所有CelebA元数据文本文件（属性列表、划分文件这些）也放到$dataroot根目录，不要留在celeba子目录里

之后再设置download=False运行代码，就能正常加载数据集了。

内容的提问来源于stack exchange，提问作者Soerendip