我使用Keras库构建了一个用于图像分类的CNN模型。然而,训练过程需要好几个小时。一旦我完成了模型的训练,如何在不重新训练的情况下使用它呢?我是说,在训练完成后,我想多次使用这个模型。因为我将在Android Studio中使用我的模型。任何帮助都将不胜感激
谢谢你…
编辑
当我写这个问题时,我还不知道如何保存和加载模型,在回答中你可以看到它们的正确用法。
回答:
你可以在训练过程结束后轻松保存你的模型,使用以下命令:
model.save('my_model.h5')
之后你可以使用以下命令加载该模型:
model = load_model('my_model.h5')
更多详情请查看文档:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model