如何解决使用Keras的模型类将代码从TensorFlow 1.x迁移到Tensorflow 2.x
我有一个运行“自定义”模型的代码,看来它是使用“急切模式”构造的。当我尝试运行model.predict()函数时,出现以下错误
文件“ /home/jptalledo/.local/lib/python3.6/site-packages/tensorflow/python/keras/utils/version_utils.py”,第122行,在disallow_legacy_graph中
引发ValueError(error_msg)
ValueError:构建Model.predict
实例并启用了eager模式时,不支持以图形方式调用Model
。请在图形模式下构建Model
实例,或在启用快速模式的情况下调用Model.predict
。
python代码如下:
def nn_predict(self,img):
"""Run model prediction to classify image as EV and return its probability"""
img = cv2.resize(cv2.cvtColor(img,cv2.COLOR_BGR2RGB),self.target_image_size).astype(np.float32) / 255.0
img = np.expand_dims(img,axis=0)
with self.tf_graph.as_default():
predictions = self.nn_model.predict(img)
return predictions
问题所在的位置:predictions = self.nn_model.predict(img)
任何建议如何启用急切模式?
谢谢
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。