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

问题描述:

Iam新的keras和测试一些教程与mnist图像后,我想用我自己的数据集进行训练。数据是从0-9的数字.png图像。 我命令他们进入10个班级,每个包含100个.png图像的数字分开(所以一个文件夹为0,一个文件夹为1,一个文件夹为2等..)。如何下载本地图像集与Keras一起使用?

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

您需要使用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)