【YOLO】如何將Keras訓練的模型用於OpenCV中(.h5文件轉換成.weights文件)

文章目錄 前言 操作步驟 驗證 前言 模型訓練時採用TensorFlow + Keras框架,其參數模型輸出爲.h5文件,其中包含了yolo訓練結果中的卷積層和池化層參數。 OpenCV庫,其中深度神經網絡(dnn)模塊採用了DarkNet框架,該框架導入的模型爲.weights文件,並依賴於.cfg文件所定義的架構。所以,如果想將前面的框架訓練的模型用於OpenCV中,就需要將yolo.h5文件
相關文章
相關標籤/搜索