使用Torchvision ImageFolder加载Safebooru数据集时出现文件未找到错误
问题分析与解决方案
- 报错原因:
ImageFolder的工作逻辑是扫描目标目录下的子文件夹(此处为cartoons/1/),加载其中的实体图片文件(如jpg、png等)。你当前cartoons/1/里只有all_data.csv,没有实际图片,所以触发FileNotFoundError。 - 必须下载实体图片:你拿到的
all_data.csv里的sample_url列就是图片的下载链接,必须通过这些链接批量下载对应图片到cartoons/1/目录,ImageFolder才能正常识别。 - 操作建议:
- 编写批量下载脚本,遍历
all_data.csv的sample_url字段,将图片下载到cartoons/1/目录 - 确保下载的图片为常见图像格式(jpg/png/webp等),
ImageFolder默认支持这些格式 - 下载完成后再执行
cartoon_dataset = ImageFolder('cartoons/', transformer)即可正常创建数据集
- 编写批量下载脚本,遍历
内容的提问来源于stack exchange,提问作者FabricioG
相关产品推荐
相关产品推荐

