繼yolo-v1後,又乙個yolo-v2的配置訓練教程,希望可以幫助大家。
注意:1、測試軟體:ubuntu14.04+cuda7.5+cudnn5.0+opencv2.4.10(titanx)
2、有些您複製的終端命令如果不能在終端執行,請注意英文全形半形問題,您可以將命令輸入終端,無須複製貼上命令
修改makefile檔案:
cpu設定:gpu=0,opencv=1,cudnn=0
gpu設定:gpu=1,opencv=1,cudnn=1
注意nvcc的路徑!!! (設定自己nvcc的路徑)
終端輸入:
cd darknet
make
找到voc_label.py程式,在script裡面,轉換格式,執行:
python voc_label.py
生成train檔案:
cat 2007_* 2012_train.txt > train.txt
生成的caffemodel會存放在darknet/backup目錄下
第六部分:demo
darknet目錄下執行:
./darknet detector test cfg/voc.data cfg/tiny-yolo-voc.cfg backup/tiny-yolo-voc_1000(迭代步數自己設定啦).weights data/dog.jpg
第七部分:demo效果圖:
見部落格結尾部分
第八部分:訓練自己的資料集
資料及製作部分:
參考我的yolo-v1部落格:
咱群裡檔案有轉換成yolo格式的檔案,找一下哦
訓練部分:和第五部分是一樣的啦
效果圖如下:
任何問題**
唯一qq
2258205918(名稱
samylee) !!!
或唯一vx:samylee_csdn
yolov2 訓練自己的樣本 GPU加速
1.採集影象,做成voc格式的資料集 3.修改makelist檔案裡面的 gpu 1 cudnn 1 opencv 1 openmp 1 debug 1 4.把voc格式的資料集放在scrip裡面,然後跑voc label.py 5.cfg檔案裡的voc.name classes 1如果只有一類就為...
Yolo v2 訓練 NAN解決過程
主要參考 今天在用yolo訓練自己的資料集的時候,一開始訓練過程中的loss還很正常地在下降。結果到了幾個個batch的地方突然就出現loss nan的情況。其實到現在自己也沒有發現到底是什麼導致了nan,但是參考上面這篇文章,至少暫時解決了這個問題。1 首先檢查了 是不是對的。一般出現nan,第一...
YOLOv2訓練自己的資料來實現人臉檢測詳細步驟
一 模仿voc的格式 建立資料夾vocdevkit,在裡面新建voc2017資料夾,然後voc2017裡新建下圖的4個資料夾,最後在imagesets裡新建名叫main資料夾。其中,annotations裡面放每張的.xml檔案。imagesets的main裡放train.txt,該文字檔案裡面的內...