.h5文件转.pb文件报错：NameError: name 'yolo_head' is not defined

最近在python环境下使用keras框架训练好了自己的tiny yolov3模型，得到了包含模型结构和权重等全部信息的.h5文件，打算把.h5文件转换为.pb文件，拿到C++去用，在网上找各种.h5转.pb文件的程序，结果总是碰到
NameError: name ‘yolo_head’ is not defined
查找相关文件位置，yolo_head这个函数明明存在

后来又踩了很多坑，好在最后总算解决了，下面是我的解决方案，希望可以帮助大家。

问题原因：

h5转pb时，网上程序大部分要用到load_model这个函数，而使用load_model时只能加载keras中默认已经存在的网络结构，但是tiny yolov3网络结构中含有自定义的部分结构，无法使用load_model加载。

解决方法：

先创建出一个空的yolo model出来 然后再加载你保存的权重
（使用相关函数记得import一下）

h5_model = tiny_yolo_body(Input(shape=(None, None, 3)), num_anchors//3, num_classes)  
h5_model.load_weights(weight_file_path, by_name=True, skip_mismatch=True)

然后使用网上的程序，转.pb文件就能成功了。

h5_to_pb(h5_model, output_dir, model_name)

有问题评论留言，可以协助解决。