Keras加载model出错怎么办?

2021-08-04 16:16:15 浏览数 (3)

Keras是一个出名的机器学习框架,我们用这个框架训练一个model可以用来进行预测或者一些学习操作。但是在使用这个model的时候有时候会报错,比如有时候Keras加载model的时候就会报错。这种情况常见于反复调用的情况。那么如何解决Keras加载model报错呢?接下来的这篇文章告诉你。

问题:

python+Flask 反复调用model报错

ValueError: Tensor Tensor("dense_2/Softmax:0"...) is not an element of this graph.

原因:

不明

解决方案:

在初始化加载模型之后,就随便生成一个向量让 model 执行一次 predict 函数

from keras.models import load_model
# ...
model = load_modle('xxx.hdf5')
model.predict(...)

补充:from models import * 报错,找不到models的解决办法

解决方法

from .models import *

在models 前加个点,引用当前目录下的models

以上为个人经验,希望能给大家一个参考,也希望大家多多支持W3Cschool


0 人点赞