用keras训练好模型,再在django初始化加载模型,这个过程没有问题,但是在调用到模型执行model.predict()的时候就报错:
raise ValueError("Tensor %s is not an element of this graph." % obj) ValueError: Tensor Tensor("dense_2/Softmax:0", shape=(?, 2), dtype=float32) is not an element of this graph.
解决方法:
在django中加载模型后,先执行一次model.predict()操作,之后的调用就不会出问题了
model = load_model(filepath=model_path) y = model.predict(x)
参考:https://zhuanlan.zhihu.com/p/27101000
原文地址:https://www.cnblogs.com/bymo/p/8955009.html
时间: 2024-10-08 11:57:04