yolov2 訓練自己的樣本 GPU加速

2021-08-14 03:28:10 字數 931 閱讀 2075

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如果只有一類就為1

train  = /home/******/yolov2/darknet-master/scripts/2007_train.txt

valid  = /home/******/yolov2/darknet-master/scripts/2007_val.txt

names = data/voc.names

backup = backup

6.cfg/yolo-voc.2.0.cfg

[convolutional]

size=1

stride=1

pad=1

filters=30(1類的情況)

activation=linear

[region]

anchors = 1.08,1.19,  3.42,4.41,  6.63,11.38,  9.42,5.11,  16.62,10.52

bias_match=1

classes=1(1類的情況)

coords=4

num=5

softmax=1

jitter=.2

rescore=1

7.在detector.c 加上#define gpu 1

8.cd 到darknet-master,然後make

9.訓練資料:./darknet detector train cfg/voc.data cfg/yolo-voc.2.0.cfg

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

繼yolo v1後,又乙個yolo v2的配置訓練教程,希望可以幫助大家。注意 1 測試軟體 ubuntu14.04 cuda7.5 cudnn5.0 opencv2.4.10 titanx 2 有些您複製的終端命令如果不能在終端執行,請注意英文全形半形問題,您可以將命令輸入終端,無須複製貼上命令 ...

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