Yolo v2訓練voc及自己的資料集

2021-07-25 01:44:10 字數 1010 閱讀 4320

繼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,該文字檔案裡面的內...