一、问题描述:
最近在学习一个TF2.x版本的yolov4项目,当我看明白主程序的整个检测过程以及一些网络搭建的代码之后,就准备将模型保存成.pb或者.h5格式,然后在其他程序中直接调用进行推理。保存的时候没有出现问题,成功的保存成了.pb文件,但是在使用tf.keras.models.load_model加载模型的时候,出现了name tf is not defined的报错。
保存、加载以及推理的大概流程如下:
self.yolo_model.save('./saved_model')
model=tf.keras.models.load_model('./saved_model'
