CNN训练模型未正确保存

时间:2020-06-19 15:16:07

标签: python tensorflow flask keras backend

我是使用tensorflow的新手,我建立了CNN,该CNN经过训练和测试均具有80-85%的准确度。我尝试使用model.save('example.h5')保存训练有素的模型,并使用files.download('example.h5')

下载文件

然后,我尝试使用model = tf.keras.models.load_model('example.h5')将其加载到烧瓶后端。

当我尝试将其与随机图像一起使用时,就好像该模型从未被训练过。有什么办法吗?谢谢

1 个答案:

答案 0 :(得分:0)

一个常见的原因可能是您正在对训练中的数据进行标准化,而不是在进行预测之前。如果您已经对归一化数据进行了训练,请确保在运行预测之前对输入数据进行归一化。