Pytorch中torchvision.datasets.ImageFolder報錯:Found 0 files in subfolders of:XXX,完美解決!

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.總結

在平時進行實驗時,雖然說對數據集的讀取方面可能不會有太多問題,但是還是要認真細緻一點,避免出現不必要的問題,這樣也能提高工作的效率。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章