如何在Keras中使用训练模型预测输入图像?

时间:2016-11-02 14:31:56

标签: python tensorflow deep-learning keras

我正在使用Keras和tensorflow后端构建CNN模型。我已经训练了模型6个小时。现在,我想使用我的模型预测我的自定义外部图像。如何在Keras做到这一点?谢谢

1 个答案:

答案 0 :(得分:5)

为简化起见,您可以说使用模型时通常有4个主要步骤(不仅仅是在Keras中):

  • 构建模型并对其进行编译:model.compile()
  • 使用您的训练数据训练模型:model.fit()
  • 使用您的测试数据评估模型:model.evaluate()
  • 使用您的目标数据进行预测:model.predict()

根据您需要的输出,您必须使用model.predict_proba()或model.predict_classes()。

请参阅https://keras.io/models/sequential/#sequential-model-methods以获取完整的参考和论据。

Keras存储库也有很多例子:https://github.com/fchollet/keras/tree/master/examples