yolo3,模型訓練流程

2021-09-29 13:22:08 字數 874 閱讀 2761

訓練前準備:

1,準備資料集

和邊界矩形txt

邊界矩形txt內容

類別從0開始

labeltext = str(int(labstr)-1) + ' ' + str(x) + ' ' + str(y) + ' ' + str(w) + ' ' + str(h) + '\n'

x,y,w,h為目標的矩形中心(x,y),wh目標寬高

2,修改cfg檔案

[convolutional]

size=1

stride=1

pad=1

filters=27                    修改為(cls+5)*3

activation=linear

[yolo]

mask = 0,1,2

anchors = 75,45, 87,56, 145,50, 171,136, 194,91, 240,116, 246,94, 278,53, 314,150           修改boxes的kmean

classes=4                修改類別

3,建立.names檔案,內容為分類名字

4,建立.data檔案

例如:classes= 4

train  = trainxx.txt

valid  = testxx.txt

names = data/xx.names

backup = mod/xx

5,資料集

樣本txt整理,如上面的trainxx.txt

6,執行命令

darknet detector train xx.data data/yolov3-spp-xx.cfg data/darknet53.conv.74

使用yolo3模型訓練自己的資料集

1.ubuntu16.04 2.tensorflow gpu 1.4.0 或更高版本 3.keras 2.2.4 4.numpy 1.15.2 實測1.16.1會報錯 1.使用voc2007資料集的檔案結構 注 資料集中沒有test.py,你需要將其拷貝到voc2007資料夾下。2.標註 使用lab...

YOLO3多gpu訓練方法

版本可參考git上的keras yolo3的實現,現在我手裡有兩塊gpu,所以設定多塊gpu來加快訓練速度 1.訓練 前引入 from keras.utils import multi gpu model2.找到自己構建網路的地方,第一次構建model是下面的語句 is tiny version l...

yolo3演算法講解

網路結構 假設輸入資料的大小為416 416,經過darknet 53特徵提取網路後得到尺度為13 13,26 26,52 52的特徵圖。其中每乙個尺寸的特徵圖都有18 3 6,其中3是指每個grid cell有3個anchor,每個anchor 乙個邊界框的4個座標值,1個置信度,還有乙個類別概率...