【YOLO】如何将Keras训练的模型用于OpenCV中(.h5文件转换成.weights文件)

文章目录 前言 操作步骤 验证 前言 模型训练时采用TensorFlow + Keras框架,其参数模型输出为.h5文件,其中包含了yolo训练结果中的卷积层和池化层参数。 OpenCV库,其中深度神经网络(dnn)模块采用了DarkNet框架,该框架导入的模型为.weights文件,并依赖于.cfg文件所定义的架构。所以,如果想将前面的框架训练的模型用于OpenCV中,就需要将yolo.h5文件
相关文章
相关标签/搜索