Keras加载彩色图像

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-image等加载图像（L58）
2. 调整图像尺寸和/或裁剪尺寸以适合输入尺寸（VGG-16为224 * 224）（L58）
3. 计算并减去每个图像的平均值（L59~L61）
4. 在高度和宽度尺寸（L62）之前交换颜色尺寸
5. 如果您使用scikit-image，则需要交换3个颜色通道，因为OpenCV将图像加载为 BGR 通道，但scikit-image将其加载为 RGB 通道。
6. 添加批量大小维度（L63）
7. 随机播放，分区并连接它们（沿着0号方向）以形成训练数据X_train，测试数据X_test等以及基础事实Y_train，Y_test等
8. 如果您的数据集太大而无法放入内存，请使用生成器和函数fit_generator来进行培训。 （Keras也有evaluate_generator和predict_generator）
9. 您现在已准备好接受培训。