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. 你现在已经准备好接受培训了。