model.predict_classes vs model.predict_generator in keras



我知道prection_generator输出概率。为了获得课程,我只是找到最大概率的索引,这将是最可能的班级。但是,我发现在执行此操作之后,我获得的输出与呼叫predive_classes的输出不同。我不懂为什么。有人可以解释一下吗?

keras中的生成器使用glob列出字母顺序排序的文件夹,您可以在训练期间使用

在训练期间使用 使用
# save classes to JSON
class_json = json.dumps(train_generator.class_indices)
with open("class.json", "w") as class_file:
    class_file.write(class_json)

样品在批处理生成器中随附(此处(,以便给出fit_generator或evaluate_generator的批处理时。

如https://github.com/fchollet/keras/issues/3477

所述,如果在图像上完成此操作的另一种可能性是不使用cescale =1。/255。

希望有帮助!

最新更新