如何下载本地图像集以与 Keras 一起使用？

【问题标题】：How to download local image sets to use with Keras?如何下载本地图像集以与 Keras 一起使用？
【发布时间】：2017-10-06 20:18:02
【问题描述】：

我是 keras 的新手，在使用 mnist 图像测试了一些教程之后，我想使用自己的数据集进行训练。数据是 0-9 数字的 .png 图像。我将它们分为 10 个类，每个类分别包含 100 个数字的 .png 图像（因此一个文件夹为 0，一个文件夹为 1，一个文件夹为 2，等等）。

现在我想知道如何使用 python 加载图像，以便 keras 使用它们？

【问题讨论】：

标签： python load deep-learning keras

【解决方案1】：

您需要使用 Keras 的 ImageDataGenerator().flow_from_directory() 从您的文件系统中生成批量图像数据，然后您将在这些数据上训练您的模型。在文件系统中组织好图像后，下一步就是创建ImageDataGenerator()。

This video 演示如何准备您的图像数据并创建您的ImageDataGenerator()，然后this video 演示如何在图像数据上训练您的 CNN。

这样的一个例子看起来像

train_batches = ImageDataGenerator().flow_from_directory(directory=<path_to_image_data>, target_size=(224,224), classes=[‘0’, '1', ‘2’, ‘3’, …, ‘9’], batch_size=10)

【讨论】：