我使用Keras库为图像分类建立了CNN模型。但是培训需要很多小时。训练模型后,如何在不进行训练的情况下使用它?我的意思是说,在训练模型后,我想多次使用它。 因为我将在android studio中使用我的模型。 感谢您的帮助
谢谢
答案 0 :(得分:2)
在训练过程之后,您可以使用以下方法轻松保存模型:
model.save('my_model.h5')
您以后可以使用以下方式加载该模型:
model = load_model('my_model.h5')
有关更多详细信息,请参阅文档:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model