我正在处理以下链接 https://becominghuman.ai/building-an-image-classifier-using-deep-learning-in-python-totally-from-a-beginners-perspective-be8dbaf22dd8 我怎么知道“result [0] [0]”中的“one”是对“狗”而不是猫的预测?
答案 0 :(得分:0)
本文中的代码使用ImageDataGenerator
类和flow_from_directory()
函数来加载和扩充图像。检查Keras documentation,您可以找到有关其中一个名为classes
的参数的信息:
类子目录的可选列表(例如[' dogs',' cats'])。 默认值:无。如果未提供,则列表将为 从子目录名称/结构下自动推断出 目录,其中每个子目录将被视为不同的 class(以及将映射到标签的类的顺序 索引,将是字母数字)。包含映射的字典 从类名到类索引可以通过属性获得 class_indices。
根据这些信息,我怀疑是“猫”。是指一只0而一只狗#39;指的是1.确保你可以检查班级词典:
print(train_datagen.class_indices)