1.報錯如下:
RuntimeError: Found 0 files in subfolders of:XXX
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,webp
這個錯誤一般發生在調用torchvision.datasets.ImageFolder訓練模型讀取數據時,主要原因是數據存儲路徑與數據讀取路徑不一致。
2.解決方法:
如在使用神經網絡模型處理貓狗數據集時,此時的數據集爲大量的貓狗圖片,且數據集存儲路徑爲:"/文件夾/Dog.0.jpg"或"/文件夾/Cat.1.jpg",則此時的數據讀取路徑應該爲:"/文件夾"。否則將會報錯。
3.總結
在平時進行實驗時,雖然說對數據集的讀取方面可能不會有太多問題,但是還是要認真細緻一點,避免出現不必要的問題,這樣也能提高工作的效率。