项目场景:
RuntimeError: Found 0 files in subfolders of: face_a_aligned
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp
问题描述
这是用torchvision.datasets.ImageFolder()时出现的问题
解决方案:
命令行训练时:
将文件路径设为数据集目录/train的上一级,如路径为myDataset/train,你就设置为myDataset。
注意myDataset目录下只有train这一个文件夹
如果数据集目录已经与train.py同级,可以新建一个文件夹,比如取名train01,然后将train文件夹整个剪切到新文件夹train01下:train01/train,然后将路径设置为train01pycharm训练时:
路径名改为相对路径,加单引号就行