如何在我們訓練cnn模型時加載ImageNet？

Question

我已經下載了ImageNet並解壓縮文件。因此，圖像基於類別存儲在1000個子文件夾中。

我該如何加載數據集？逐一閱讀圖像？

Answer 1

我的建議是創建一個預處理腳本，該腳本從該子文件夾中讀取所有圖像，並將其集成到將保留到磁盤的唯一結構中，例如寫入磁盤或使用pickle的numpy對象。