一、问题描述:
最近在学习一个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'