相关疑难解决方法(0)

打开使用 Tensorflow 后端的 Keras 模型时出现 NameError

我想在我的第一个 Keras 层中调整我的输入图像的大小,所以我遵循了这个SO 问题。解决方案效果很好,直到我保存了我的模型,然后尝试在另一个文件中使用它并抛出

NameError: name 'ktf' is not defined
Run Code Online (Sandbox Code Playgroud)

我尝试添加:

from keras.backend import tf as ktf
Run Code Online (Sandbox Code Playgroud)

到打开模型的文件,但它仍然无法在模型中识别它。我需要做什么才能让打开保存模型的程序识别 tensorflow 后端使用的函数?


还有一些细节...

火车.py:

from keras.backend import tf as ktf

#Other stuff...

model = Sequential()
model.add(Lambda(lambda x: ktf.image.resize_images(x, (80, 160)), input_shape=(160, 320, 3))) #This line referenced in error

#Rest of model and training...

model.save('model.h5')
Run Code Online (Sandbox Code Playgroud)

评估.py:

from keras.backend import tf as ktf

#Other stuff...

model = load_model('model.h5') #Error is here
Run Code Online (Sandbox Code Playgroud)

错误信息:

Using TensorFlow backend.
Traceback (most recent call last):
  File "C:\program\eval.py", …
Run Code Online (Sandbox Code Playgroud)

python keras tensorflow

2
推荐指数
1
解决办法
1万
查看次数

标签 统计

keras ×1

python ×1

tensorflow ×1