前言
模型训练时采用TensorFlow + Keras框架,其参数模型输出为.h5文件,其中包含了yolo训练结果中的卷积层和池化层参数。
OpenCV库,其中深度神经网络(dnn)模块采用了DarkNet框架,该框架导入的模型为.weights文件,并依赖于.cfg文件所定义的架构。所以,如果想将前面的框架训练的模型用于OpenCV中,就需要将yolo.h5文件转换为yolo.weights文件。
关于Keras训练YOLO模型的方法,可参考:https://blog.csdn.net/qinchang1/article/details/89608058
关于采用OpenCV进行YOLO检测的方法,可参考:https://blog.csdn.net/qinchang1/article/details/89435023
操作步骤
程序代码可以通过下面的链接下载:
百度云链接:https://pan.baidu.com/s/1-yX67ArLq9cFwCYGLTkPHg 提取码:dhvw
CSDN下载: