如何下载本地图像集与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)