我想在我的第一个 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)