【keras自定义上采样层】NameError: name ‘tf‘ is not defined

在unet网络中自定义了上采样函数,但是在预测导入模型的时候却报错了

NameError: name 'tf' is not defined

原因是需要在导入预测文件中导入模型的时候传入相应参数

model = keras.models.load_model('my_model.h5', custom_objects={'tf': tf})

当然对应的需要使用import tensorflow as tf

使用keras.models.load_model来导入模型,那么在训练保存模型的时候用m.save

【keras自定义上采样层】NameError: name ‘tf‘ is not defined_第1张图片

你可能感兴趣的:(debug,keras,深度学习)