我已经将我的整个数据分离到不同类别的不同文件夹中,用于手数字识别。如何让我的模型知道某个类a包含描述字母a的图像。
train_datagen = ImageDataGenerator(
rescale=1.0 / 255, shear_range=0.2, zoom_range=0.2, horizontal_flip=True
)
train_generator = train_datagen.flow_from_directory(
"Image Folder/", target_size=(28, 28), batch_size=1, class_mode="binary"
)
print(train_generator[0])
在上面的例子中,Image folder
必须为每个类都有子文件夹,如下所示:
- 一些/路径/
- class1/
- image1.jpg
- image2.jpg
- 类别2/
- image3.jpg
- 等等
- 等等
- class1/