記錄,基於keras的YOLOv3的實現和解讀

2021-09-24 18:44:43 字數 1205 閱讀 9926

本人正在使用yolov3進行車流量的偵測,記錄一下學習的過程,使用的開源**為keras-yolo3-master 鏈結如下:keras-yolo3-master

本人從使用角度及**角度入手,關於yolov3的解釋請參考yolov3

if __name__ == '__main__':

detect_video(yolo(),'d:\pycharmprojects\carflowdetection\dete.mp4')

train.py

yolo.py

model.py

暫時先說說對這三個檔案的理解:這是自己對該**見解,若有錯誤請批評指教

這幾行**主要指定了訓練的類別檔案、anchor檔案的路徑。anchors的值可以使用預設的也可以通過執行kmeans.py生成

通過anchors的個數對使用的模型進行判斷,如果為6則建立tiny_yolov3,否則建立yolov3.

這部分主要是對模型編譯和訓練的引數進行設定,具體有些引數還在理解當中。

這個階段的訓練主要是得到乙個穩定的loss,為第二階段的訓練做準備。

第二階段訓練,fine-tune引數,使得模型變得更好。最後儲存訓練的模型為trained_weights_final.h5

這些函式主要是讀取classname,anchor,以及訓練集影象和box

train.py檔案主要完成的就是載入必要的東西,編譯網路並對網路進行訓練

今天先寫到這裡,未完待續。。。

基於Tensorflow的Keras安裝

平台 ubuntu14.04 版本 python3.5,anaconda3 4.1.1,tensorflow1.4.0 bash anaconda sh 檔名 步驟2 回到剛開啟命令框時的目錄,依此輸入 mkdir pip cd pip vim pipconfig 然後會彈出一些配置 拉到最下面,點...

Keras下實現yolov2並使用自己的資料集

在windows下想嘗試弄個yolo試試,先從v2版本開始吧,本文主要記錄復現 出現的問題,以及使用自己資料庫測試出現的問題。一.復現 環境 windows10下使用anaconda pycharm 顯示卡1080ti 為了用gpu肯定是conda install keras gpu,一定要保證有c...

yolov3檢測人頭 基於yolo v3的人臉檢測

一 實驗環境的搭建 1.安裝環境配置 python 3.6 tensorflow gpu 1.6.0 keras pillow opencv python matplotlib numpy 沒有通過pip安裝 例如 pip install keras 2.啟動虛擬環境,並且安裝對應的實驗環境 3.模...