1、在https://github.com/qqwweee/keras-yolo3 下載yolo-v3的算法
2呕屎、在https://pjreddie.com/media/files/yolov3.weights下載權(quán)重并放在keras-yolo3的文件夾下
3诞挨、配置cuda環(huán)境
????下載電腦適配的cuda览露,cudnn版本逃顶,兩個(gè)版本有對(duì)應(yīng)起來(lái)豆挽。(本人cuda8.0托呕,cudnn6.0)
3含蓉、配置電腦python環(huán)境
(1)安裝anaconda
(2)安裝keras庫(kù) pip install keras
(2)安裝tensorflow gpu版本 pip tensorflow-gpu==1.01(測(cè)試貌似只有這個(gè)版本能跟cuda8.0對(duì)應(yīng)起來(lái))
4、python convert.py 將darknet類中的權(quán)重轉(zhuǎn)化為keras中要用的類型 .h5文檔项郊。
5馅扣、python yolo.py 進(jìn)行.jpg檢測(cè)
6、python vedio_yolo.py進(jìn)行視頻檢測(cè)或者調(diào)用攝像頭(需要opencv )
7着降、訓(xùn)練自己的數(shù)據(jù)集
(1)將自己的數(shù)據(jù)做成VOC格式差油,包含原JPEG圖像和用labeling標(biāo)注生成的XML文檔集,修改cfg,voc_name蓄喇,文檔发侵。
(2)寫個(gè)簡(jiǎn)單的腳本提取所有的JPEG文檔名,做成train.txt 放在主文件夾下
(3)python voc_annotation.py生成yolov3訓(xùn)練所需的txt文檔
(4)python train.py
warning:可能遇到的錯(cuò)誤?
? ? ? ? ? ? ? ? (1)內(nèi)存泄露錯(cuò)誤妆偏,顯存小的情況下要降低batch_size的大小
? ? ? ? ? ? ? ? (2)relu函數(shù)問(wèn)題 刃鳄,按照網(wǎng)上的自己寫一個(gè)relu函數(shù)就可以了。