Keras loading彩色圖像

Question

我有3個彩色圖像文件夾。文件夾的名稱是內部圖像的標籤。

cls1 
    |____img_0.png 
    |____ ... 
    |____img_n.png 
cls2 
    |____img_0.png 
    |____ ... 
    |____img_n.png 
cls3 
    |____img_0.png 
    |____ ... 
    |____img_n.png 

我想用Keras庫創建分類卷積神經網絡，但我找不到，如何創建彩色圖像數據集。 你能幫我嗎？

Answer 1

請考慮這個gist of pre-trained VGG-16 model with example usage我覺得這說明：

總結：

1. 負載OpenCV的或scikit，圖像等您的圖像（L58）
2. 調整大小和/或裁剪你圖像以適合輸入尺寸（對於VGG-16爲224 * 224）（L58）
3. 計算併爲每幅圖像減去均值（L59〜L61）
4. 將色彩維度高度和寬度尺寸（L62）
5. 如果使用scikit圖像，需要交換的3個顏色通道的OpenCV因爲負載的圖像作爲BGR通道但scikit圖像加載它作爲RGB通道。
6. 添加批量大小尺寸（L63）
7. 洗牌，分區和將它們連接起來（沿着維度0），以形成訓練數據X_train，測試數據等X_test隨着地面實況Y_train，Y_test等
8. 如果您的數據集太大而不適合內存，請使用生成器並使用功能fit_generator來進行培訓。 （凱拉斯也有evaluate_generator和predict_generator）
9. 你現在已經準備好接受培訓了。