在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