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

Question

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

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

Answer 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)