Keras加载model出错怎么办?
2021-08-04 16:16:15
浏览数 (3547)
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。