如何为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
    • 类别2/
      • image3.jpg
      • 等等
    • 等等

相关内容

  • 没有找到相关文章

最新更新