如何正确构造数据以让ML模型解释它

问题描述

我已将我的整个数据分为不同的文件夹，用于不同类别的手手指识别。如何让我的模型知道某个类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
  • class2 /
    • image3.jpg
    • 等
  • 等