keras中的model.predict_classes与model.predict_generator

Question

我知道predict_generator输出概率。为了得到这个类，我找到了概率最大的索引，这就是最可能的类。然而，我发现在这样做之后，我得到的输出与调用predict_classes时不同。我不明白为什么。有人能解释一下这个吗？

Answer 1

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中提到的ImageDataGenerator中的rescale=1./255

希望对大家有所帮助！