主頁 >  其他 > YOLO V4的模型訓練

YOLO V4的模型訓練

2020-09-12 14:23:33 其他

1、YOLO V4模型訓練的基本思路

  所有機器學習涉及模型訓練,一般都有訓練集、驗證集、測驗集,因此需要準備資料集,有了資料集,再呼叫訓練的演算法,獲取訓練的結果,v3、v4模型訓練方法相同,

 

2、YOLO V4模型訓練的體驗

  利用已有資料,體驗一下模型訓練的各個步驟,

  網址:https://pjreddie.com/darknet/yolo/給出了模型訓練的方法,

 

2.1、YOLO模型訓練的資料集格式

  YOLO訓練所要求的資料格式是PASCAL VOC或者COCO等標準資料集格式,

  darknet\build\darknet\x64\data\voc目錄下,有一個voc_label.py檔案,用于資料轉換,涉及的資料集的格式為PASCAL VOC格式,

  關于PASCAL VOC格式的介紹,參見:

  • 《PASCAL VOC資料集的標注格式》(https://zhuanlan.zhihu.com/p/33405410);

  • 《計算機視覺標準資料集整理—PASCAL VOC資料集》(https://blog.csdn.net/xingwei_09/article/details/79142558);

  • 《VOC 2007資料集結構》(https://blog.csdn.net/qq_38273984/article/details/90749314

 

  關于Annotations的xml檔案中,影像size的depth屬性,含義為通道數,RGB影像,值為3,參見:https://blog.csdn.net/qq_18343569/article/details/47830503

 

2.2、構造VOC目錄結構

  按照上面參考網文的說法,構造基本的目錄結構:Annotations、JPEGImages以及ImageSets/Main,

  我這邊正好搞到了VOC2012的訓練/驗證的資料集,

  如果要下載,地址:https://pjreddie.com/projects/pascal-voc-dataset-mirror/,檔案都點大,

  先將VOC2012資料集復制到darknet\build\darknet\x64\data\voc目錄下,結構如下:

darknet\build\darknet\x64\data\voc\VOCdevkit
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2012
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2012\...
...

 

2.3、執行voc_label.py檔案

  voc_label.py檔案的決議,可參考:《YOLO_DarkNet_決議之旅_voc_label.py的決議》(https://blog.csdn.net/Willen/article/details/83868164),

  由于我使用VOC2012資料集,沒有VOC2007,ImageSets/Main中,只有train,val和trainval檔案,因此,需要修改voc_label.py檔案,

#line7 修改
sets=[('2012', 'train'), ('2012', 'val'), ('2007', 'train'), ('2007', 'val'), ('2007', 'test')]
#==>
sets=[('2012', 'train'), ('2012', 'val'),('2012', 'trainval')]
?

  #否則,報錯:

FileNotFoundError: [Errno 2] No such file or directory: 'VOCdevkit/VOC2007/ImageSets/Main/train.txt'

 

  然后,執行voc_label.py檔案,在AnaConda環境下,執行下列陳述句:

(tensorflow) E:\workspace\darknet\build\darknet\x64\data\voc>python voc_label.py

  獲取下列檔案:

darknet\build\darknet\x64\data\voc\2012_train.txt
darknet\build\darknet\x64\data\voc\2012_val.txt
darknet\build\darknet\x64\data\voc\2012_trainval.txt
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2012\labels目錄及檔案

 

2.4、模型訓練

  首先,修改darknet/build/darknet/x64/data中的voc.data,修改train和valid這兩行的資料檔案路徑,

classes= 20
train = data/voc/2012_train.txt
valid = data/voc/2012_val.txt
#difficult = data/difficult_2007_test.txt
names = data/voc.names
backup = backup/

  然后執行下列陳述句:

darknet.exe detector train data/voc.data cfg/yolov4.cfg yolov4.weights

  最后一個引數為預訓練權重檔案,應該使用只包含卷積層的預訓練權重檔案,下載地址:https://pjreddie.com/media/files/darknet53.conv.74

  我這里擔心訓練時間過長,偷懶了,直接用yolov4.weights,

 

  運行結果如下:

E:\workspace\darknet\build\darknet\x64>darknet.exe detector train data/voc.data cfg/yolov4.cfg yolov4.weights
GPU isn't used
Used AVX
Used FMA & AVX2
OpenCV version: 4.2.0
yolov4
mini_batch = 8, batch = 64, time_steps = 1, train = 1
  layer   filters size/strd(dil)     input               output
  0 conv     32       3 x 3/ 1   608 x 608 x   3 -> 608 x 608 x 32 0.639 BF
  1 conv     64       3 x 3/ 2   608 x 608 x 32 -> 304 x 304 x 64 3.407 BF
  2 conv     64       1 x 1/ 1   304 x 304 x 64 -> 304 x 304 x 64 0.757 BF
  3 route 1                                     -> 304 x 304 x 64
  4 conv     64       1 x 1/ 1   304 x 304 x 64 -> 304 x 304 x 64 0.757 BF
  5 conv     32       1 x 1/ 1   304 x 304 x 64 -> 304 x 304 x 32 0.379 BF
  6 conv     64       3 x 3/ 1   304 x 304 x 32 -> 304 x 304 x 64 3.407 BF
  7 Shortcut Layer: 4, wt = 0, wn = 0, outputs: 304 x 304 x 64 0.006 BF
  8 conv     64       1 x 1/ 1   304 x 304 x 64 -> 304 x 304 x 64 0.757 BF
  9 route 8 2                                   -> 304 x 304 x 128
10 conv     64       1 x 1/ 1   304 x 304 x 128 -> 304 x 304 x 64 1.514 BF
11 conv   128       3 x 3/ 2   304 x 304 x 64 -> 152 x 152 x 128 3.407 BF
12 conv     64       1 x 1/ 1   152 x 152 x 128 -> 152 x 152 x 64 0.379 BF
13 route 11                                     -> 152 x 152 x 128
14 conv     64       1 x 1/ 1   152 x 152 x 128 -> 152 x 152 x 64 0.379 BF
15 conv     64       1 x 1/ 1   152 x 152 x 64 -> 152 x 152 x 64 0.189 BF
16 conv     64       3 x 3/ 1   152 x 152 x 64 -> 152 x 152 x 64 1.703 BF
17 Shortcut Layer: 14, wt = 0, wn = 0, outputs: 152 x 152 x 64 0.001 BF
18 conv     64       1 x 1/ 1   152 x 152 x 64 -> 152 x 152 x 64 0.189 BF
19 conv     64       3 x 3/ 1   152 x 152 x 64 -> 152 x 152 x 64 1.703 BF
20 Shortcut Layer: 17, wt = 0, wn = 0, outputs: 152 x 152 x 64 0.001 BF
21 conv     64       1 x 1/ 1   152 x 152 x 64 -> 152 x 152 x 64 0.189 BF
22 route 21 12                                 -> 152 x 152 x 128
23 conv   128       1 x 1/ 1   152 x 152 x 128 -> 152 x 152 x 128 0.757 BF
24 conv   256       3 x 3/ 2   152 x 152 x 128 ->   76 x 76 x 256 3.407 BF
25 conv   128       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 128 0.379 BF
26 route 24                                     ->   76 x 76 x 256
27 conv   128       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 128 0.379 BF
28 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
29 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
30 Shortcut Layer: 27, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
31 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
32 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
33 Shortcut Layer: 30, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
34 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
35 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
36 Shortcut Layer: 33, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
37 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
38 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
39 Shortcut Layer: 36, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
40 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
41 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
42 Shortcut Layer: 39, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
43 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
44 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
45 Shortcut Layer: 42, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
46 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
47 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
48 Shortcut Layer: 45, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
49 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
50 conv   128       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 128 1.703 BF
51 Shortcut Layer: 48, wt = 0, wn = 0, outputs: 76 x 76 x 128 0.001 BF
52 conv   128       1 x 1/ 1     76 x 76 x 128 ->   76 x 76 x 128 0.189 BF
53 route 52 25                                 ->   76 x 76 x 256
54 conv   256       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 256 0.757 BF
55 conv   512       3 x 3/ 2     76 x 76 x 256 ->   38 x 38 x 512 3.407 BF
56 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
57 route 55                                     ->   38 x 38 x 512
58 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
59 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
60 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
61 Shortcut Layer: 58, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
62 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
63 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
64 Shortcut Layer: 61, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
65 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
66 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
67 Shortcut Layer: 64, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
68 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
69 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
70 Shortcut Layer: 67, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
71 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
72 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
73 Shortcut Layer: 70, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
74 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
75 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
76 Shortcut Layer: 73, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
77 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
78 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
79 Shortcut Layer: 76, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
80 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
81 conv   256       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 256 1.703 BF
82 Shortcut Layer: 79, wt = 0, wn = 0, outputs: 38 x 38 x 256 0.000 BF
83 conv   256       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 256 0.189 BF
84 route 83 56                                 ->   38 x 38 x 512
85 conv   512       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 512 0.757 BF
86 conv   1024       3 x 3/ 2     38 x 38 x 512 ->   19 x 19 x1024 3.407 BF
87 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
88 route 86                                     ->   19 x 19 x1024
89 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
90 conv   512       1 x 1/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.189 BF
91 conv   512       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x 512 1.703 BF
92 Shortcut Layer: 89, wt = 0, wn = 0, outputs: 19 x 19 x 512 0.000 BF
93 conv   512       1 x 1/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.189 BF
94 conv   512       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x 512 1.703 BF
95 Shortcut Layer: 92, wt = 0, wn = 0, outputs: 19 x 19 x 512 0.000 BF
96 conv   512       1 x 1/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.189 BF
97 conv   512       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x 512 1.703 BF
98 Shortcut Layer: 95, wt = 0, wn = 0, outputs: 19 x 19 x 512 0.000 BF
99 conv   512       1 x 1/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.189 BF
100 conv   512       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x 512 1.703 BF
101 Shortcut Layer: 98, wt = 0, wn = 0, outputs: 19 x 19 x 512 0.000 BF
102 conv   512       1 x 1/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.189 BF
103 route 102 87                                 ->   19 x 19 x1024
104 conv   1024       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x1024 0.757 BF
105 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
106 conv   1024       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x1024 3.407 BF
107 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
108 max               5x 5/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.005 BF
109 route 107                                           ->   19 x 19 x 512
110 max               9x 9/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.015 BF
111 route 107                                           ->   19 x 19 x 512
112 max               13x13/ 1     19 x 19 x 512 ->   19 x 19 x 512 0.031 BF
113 route 112 110 108 107                       ->   19 x 19 x2048
114 conv   512       1 x 1/ 1     19 x 19 x2048 ->   19 x 19 x 512 0.757 BF
115 conv   1024       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x1024 3.407 BF
116 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
117 conv   256       1 x 1/ 1     19 x 19 x 512 ->   19 x 19 x 256 0.095 BF
118 upsample                 2x   19 x 19 x 256 ->   38 x 38 x 256
119 route 85                                     ->   38 x 38 x 512
120 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
121 route 120 118                               ->   38 x 38 x 512
122 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
123 conv   512       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 512 3.407 BF
124 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
125 conv   512       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 512 3.407 BF
126 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
127 conv   128       1 x 1/ 1     38 x 38 x 256 ->   38 x 38 x 128 0.095 BF
128 upsample                 2x   38 x 38 x 128 ->   76 x 76 x 128
129 route 54                                     ->   76 x 76 x 256
130 conv   128       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 128 0.379 BF
131 route 130 128                               ->   76 x 76 x 256
132 conv   128       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 128 0.379 BF
133 conv   256       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 256 3.407 BF
134 conv   128       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 128 0.379 BF
135 conv   256       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 256 3.407 BF
136 conv   128       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 128 0.379 BF
137 conv   256       3 x 3/ 1     76 x 76 x 128 ->   76 x 76 x 256 3.407 BF
138 conv   255       1 x 1/ 1     76 x 76 x 256 ->   76 x 76 x 255 0.754 BF
139 yolo
[yolo] params: iou loss: ciou (4), iou_norm: 0.07, cls_norm: 1.00, scale_x_y: 1.20
nms_kind: greedynms (1), beta = 0.600000
140 route 136                                           ->   76 x 76 x 128
141 conv   256       3 x 3/ 2     76 x 76 x 128 ->   38 x 38 x 256 0.852 BF
142 route 141 126                               ->   38 x 38 x 512
143 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
144 conv   512       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 512 3.407 BF
145 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
146 conv   512       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 512 3.407 BF
147 conv   256       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 256 0.379 BF
148 conv   512       3 x 3/ 1     38 x 38 x 256 ->   38 x 38 x 512 3.407 BF
149 conv   255       1 x 1/ 1     38 x 38 x 512 ->   38 x 38 x 255 0.377 BF
150 yolo
[yolo] params: iou loss: ciou (4), iou_norm: 0.07, cls_norm: 1.00, scale_x_y: 1.10
nms_kind: greedynms (1), beta = 0.600000
151 route 147                                           ->   38 x 38 x 256
152 conv   512       3 x 3/ 2     38 x 38 x 256 ->   19 x 19 x 512 0.852 BF
153 route 152 116                               ->   19 x 19 x1024
154 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
155 conv   1024       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x1024 3.407 BF
156 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
157 conv   1024       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x1024 3.407 BF
158 conv   512       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 512 0.379 BF
159 conv   1024       3 x 3/ 1     19 x 19 x 512 ->   19 x 19 x1024 3.407 BF
160 conv   255       1 x 1/ 1     19 x 19 x1024 ->   19 x 19 x 255 0.189 BF
161 yolo
[yolo] params: iou loss: ciou (4), iou_norm: 0.07, cls_norm: 1.00, scale_x_y: 1.05
nms_kind: greedynms (1), beta = 0.600000
Total BFLOPS 128.459
avg_outputs = 1068395
Loading weights from yolov4.weights...
seen 64, trained: 32032 K-images (500 Kilo-batches_64)
Done! Loaded 162 layers from weights-file
Learning Rate: 0.0013, Momentum: 0.949, Decay: 0.0005
If error occurs - run training with flag: -dont_show
Saving weights to backup//yolov4_final.weights
Create 6 permanent cpu-threads

 

  結果在darknet\build\darknet\x64\backup目錄,生成一個yolov4_final.weights,

  訓練很快就完成了,這是因為初始權重檔案yolov4.weights,已經是根據上述資料訓練的,迭代終止條件立即滿足,可以認為:yolov4_final.weights和yolov4.weights是相同的,

 

  至此,基本的模型訓練體驗完畢,下面的作業,是結合實際應用來實作自己的模型訓練,

 

3、訓練自己的模型

3.1、建立自己的VOC目錄結構

  構造基本的VOC目錄結構:Annotations、JPEGImages以及ImageSets/Main,

  為了簡單起見,仍可使用voc_label.py檔案,我按如下方式構造目錄:

#在VOCdevkit目錄下創建VOC2020
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2020
#創建三個基本子目錄
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2020\Annotations
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2020\JPEGImages
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2020\ImageSets
#ImageSets目錄再創建Main子目錄
darknet\build\darknet\x64\data\voc\VOCdevkit\VOC2020\ImageSets\Main

 

  (下列所有修改,應該先備份一下原檔案)

 

  修改voc_label.py檔案:

sets=[('2020', 'train'), ('2020', 'val'),('2020', 'trainval'),('2020', 'test')]

 

  修改darknet\build\darknet\x64\data目錄下的voc.data檔案的對應行:

train  = data/voc/2020_train.txt
valid = data/voc/2020_val.txt

 

3.2、設定自己的分類

  關于分類,網上資料顯示,VOC資料集分類最大20個,COCO資料集最大80個,這個最大分類數可以調整為自己需要的數值,當然分類很多時,需要使用YOLO9000模型,

  參考了下列資料:

  • 《一文看懂YOLO v3》(https://zhuanlan.zhihu.com/p/60944510);

  • 《YOLO_V3 原理以及訓練說明》 (https://blog.csdn.net/qq_31511955/article/details/87917308);

  • 《YOLOv3訓練自己的資料》 (https://www.jianshu.com/p/2f1608a4459b);

  • 《每一步超詳細!制作自己的voc資料集并通過yolov3訓練》(https://blog.csdn.net/qq_34806812/article/details/81673798

  • 《一文詳解YOLO 2與YOLO 9000目標檢測系統| 分享總結》(https://baijiahao.baidu.com/s?id=1576943106747207357&wfr=spider&for=pc),

 

  假設自己的分類數為[n],則需要修改之處有:

  1)修改darknet\build\darknet\x64\data\voc.names

  將自己的[n]個分類標簽輸入,每個分類一行,

  2)修改darknet\build\darknet\x64\data\voc.data

classes= 20 ==>  classes= [n]

  3)修改darknet\build\darknet\x64\cfg\yolov4.cfg

[yolo]標簽下:
classes=80 ==> classes= [n]
?
[yolo]標簽前面的一個[convolutional]標簽:
filters=255 ==> filters=([n]+5)*3

 

3.3、標注自己的資料

  使用labelImg工具來進行標注作業,參見《windows下使用labelImg標注影像》(https://blog.csdn.net/python_pycharm/article/details/85338801),

  這是一個費時的作業,

  圖片放置在JPEGImages目錄下,最好按編號命名,對應的labelImg生成的標注檔案,放入Annotations目錄下,命名與圖片檔案名一致,

  然后,生成ImageSets/Main下的檔案,網上有makeTxt.py檔案,生成train,val,test,trainval檔案,參見:https://blog.csdn.net/public669/article/details/98020800),

  但我發現這是相當于執行了voc_label.py后,生成的檔案,暫時可以用這個方法,

  問題

  目前VOC2012/ImageSets/Main目錄下的檔案形式,是按型別分的,如car這個分類,有下列檔案:

car_train.txt
car_val.txt
car_trainval.txt

  考慮到后續處理的需要,需要一個新的處理python腳本,基本思路是:掃描分類檔案voc.names,針對每一個分類,掃描Annotations中的檔案,檢測是否包含這個分類的物件,設定[flag]為1或-1,加入這個分類的list物件,最后再打散,生成各個分類的train,val,test,trainval檔案,

 

3.4、后續作業

  類似于2.3和2.4步,執行訓練即可,我目前還沒有開始具體應用分類試驗,暫時先進行到此,

 

轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/14691.html

標籤:其他

上一篇:Python中的資料結構

下一篇:機器學習:梯度下降

標籤雲
其他(157675) Python(38076) JavaScript(25376) Java(17977) C(15215) 區塊鏈(8255) C#(7972) AI(7469) 爪哇(7425) MySQL(7132) html(6777) 基礎類(6313) sql(6102) 熊猫(6058) PHP(5869) 数组(5741) R(5409) Linux(5327) 反应(5209) 腳本語言(PerlPython)(5129) 非技術區(4971) Android(4554) 数据框(4311) css(4259) 节点.js(4032) C語言(3288) json(3245) 列表(3129) 扑(3119) C++語言(3117) 安卓(2998) 打字稿(2995) VBA(2789) Java相關(2746) 疑難問題(2699) 细绳(2522) 單片機工控(2479) iOS(2429) ASP.NET(2402) MongoDB(2323) 麻木的(2285) 正则表达式(2254) 字典(2211) 循环(2198) 迅速(2185) 擅长(2169) 镖(2155) 功能(1967) .NET技术(1958) Web開發(1951) python-3.x(1918) HtmlCss(1915) 弹簧靴(1913) C++(1909) xml(1889) PostgreSQL(1872) .NETCore(1853) 谷歌表格(1846) Unity3D(1843) for循环(1842)

熱門瀏覽
  • 網閘典型架構簡述

    網閘架構一般分為兩種:三主機的三系統架構網閘和雙主機的2+1架構網閘。 三主機架構分別為內端機、外端機和仲裁機。三機無論從軟體和硬體上均各自獨立。首先從硬體上來看,三機都用各自獨立的主板、記憶體及存盤設備。從軟體上來看,三機有各自獨立的作業系統。這樣能達到完全的三機獨立。對于“2+1”系統,“2”分為 ......

    uj5u.com 2020-09-10 02:00:44 more
  • 如何從xshell上傳檔案到centos linux虛擬機里

    如何從xshell上傳檔案到centos linux虛擬機里及:虛擬機CentOs下執行 yum -y install lrzsz命令,出現錯誤:鏡像無法找到軟體包 前言 一、安裝lrzsz步驟 二、上傳檔案 三、遇到的問題及解決方案 總結 前言 提示:其實很簡單,往虛擬機上安裝一個上傳檔案的工具 ......

    uj5u.com 2020-09-10 02:00:47 more
  • 一、SQLMAP入門

    一、SQLMAP入門 1、判斷是否存在注入 sqlmap.py -u 網址/id=1 id=1不可缺少。當注入點后面的引數大于兩個時。需要加雙引號, sqlmap.py -u "網址/id=1&uid=1" 2、判斷文本中的請求是否存在注入 從文本中加載http請求,SQLMAP可以從一個文本檔案中 ......

    uj5u.com 2020-09-10 02:00:50 more
  • Metasploit 簡單使用教程

    metasploit 簡單使用教程 浩先生, 2020-08-28 16:18:25 分類專欄: kail 網路安全 linux 文章標簽: linux資訊安全 編輯 著作權 metasploit 使用教程 前言 一、Metasploit是什么? 二、準備作業 三、具體步驟 前言 Msfconsole ......

    uj5u.com 2020-09-10 02:00:53 more
  • 游戲逆向之驅動層與用戶層通訊

    驅動層代碼: #pragma once #include <ntifs.h> #define add_code CTL_CODE(FILE_DEVICE_UNKNOWN,0x800,METHOD_BUFFERED,FILE_ANY_ACCESS) /* 更多游戲逆向視頻www.yxfzedu.com ......

    uj5u.com 2020-09-10 02:00:56 more
  • 北斗電力時鐘(北斗授時服務器)讓網路資料更精準

    北斗電力時鐘(北斗授時服務器)讓網路資料更精準 北斗電力時鐘(北斗授時服務器)讓網路資料更精準 京準電子科技官微——ahjzsz 近幾年,資訊技術的得了快速發展,互聯網在逐漸普及,其在人們生活和生產中都得到了廣泛應用,并且取得了不錯的應用效果。計算機網路資訊在電力系統中的應用,一方面使電力系統的運行 ......

    uj5u.com 2020-09-10 02:01:03 more
  • 【CTF】CTFHub 技能樹 彩蛋 writeup

    ?碎碎念 CTFHub:https://www.ctfhub.com/ 筆者入門CTF時時剛開始刷的是bugku的舊平臺,后來才有了CTFHub。 感覺不論是網頁UI設計,還是題目質量,賽事跟蹤,工具軟體都做得很不錯。 而且因為獨到的金幣制度的確讓人有一種想去刷題賺金幣的感覺。 個人還是非常喜歡這個 ......

    uj5u.com 2020-09-10 02:04:05 more
  • 02windows基礎操作

    我學到了一下幾點 Windows系統目錄結構與滲透的作用 常見Windows的服務詳解 Windows埠詳解 常用的Windows注冊表詳解 hacker DOS命令詳解(net user / type /md /rd/ dir /cd /net use copy、批處理 等) 利用dos命令制作 ......

    uj5u.com 2020-09-10 02:04:18 more
  • 03.Linux基礎操作

    我學到了以下幾點 01Linux系統介紹02系統安裝,密碼啊破解03Linux常用命令04LAMP 01LINUX windows: win03 8 12 16 19 配置不繁瑣 Linux:redhat,centos(紅帽社區版),Ubuntu server,suse unix:金融機構,證券,銀 ......

    uj5u.com 2020-09-10 02:04:30 more
  • 05HTML

    01HTML介紹 02頭部標簽講解03基礎標簽講解04表單標簽講解 HTML前段語言 js1.了解代碼2.根據代碼 懂得挖掘漏洞 (POST注入/XSS漏洞上傳)3.黑帽seo 白帽seo 客戶網站被黑帽植入劫持代碼如何處理4.熟悉html表單 <html><head><title>TDK標題,描述 ......

    uj5u.com 2020-09-10 02:04:36 more
最新发布
  • 2023年最新微信小程式抓包教程

    01 開門見山 隔一個月發一篇文章,不過分。 首先回顧一下《微信系結手機號資料庫被脫庫事件》,我也是第一時間得知了這個訊息,然后跟蹤了整件事情的經過。下面是這起事件的相關截圖以及近日流出的一萬條資料樣本: 個人認為這件事也沒什么,還不如關注一下之前45億快遞資料查詢渠道疑似在近日復活的訊息。 訊息是 ......

    uj5u.com 2023-04-20 08:48:24 more
  • web3 產品介紹:metamask 錢包 使用最多的瀏覽器插件錢包

    Metamask錢包是一種基于區塊鏈技術的數字貨幣錢包,它允許用戶在安全、便捷的環境下管理自己的加密資產。Metamask錢包是以太坊生態系統中最流行的錢包之一,它具有易于使用、安全性高和功能強大等優點。 本文將詳細介紹Metamask錢包的功能和使用方法。 一、 Metamask錢包的功能 數字資 ......

    uj5u.com 2023-04-20 08:47:46 more
  • vulnhub_Earth

    前言 靶機地址->>>vulnhub_Earth 攻擊機ip:192.168.20.121 靶機ip:192.168.20.122 參考文章 https://www.cnblogs.com/Jing-X/archive/2022/04/03/16097695.html https://www.cnb ......

    uj5u.com 2023-04-20 07:46:20 more
  • 從4k到42k,軟體測驗工程師的漲薪史,給我看哭了

    清明節一過,盲猜大家已經無心上班,在數著日子準備過五一,但一想到銀行卡里的余額……瞬間心情就不美麗了。最近,2023年高校畢業生就業調查顯示,本科畢業月平均起薪為5825元。調查一出,便有很多同學表示自己又被平均了。看著這一資料,不免讓人想到前不久中國青年報的一項調查:近六成大學生認為畢業10年內會 ......

    uj5u.com 2023-04-20 07:44:00 more
  • 最新版本 Stable Diffusion 開源 AI 繪畫工具之中文自動提詞篇

    🎈 標簽生成器 由于輸入正向提示詞 prompt 和反向提示詞 negative prompt 都是使用英文,所以對學習母語的我們非常不友好 使用網址:https://tinygeeker.github.io/p/ai-prompt-generator 這個網址是為了讓大家在使用 AI 繪畫的時候 ......

    uj5u.com 2023-04-20 07:43:36 more
  • 漫談前端自動化測驗演進之路及測驗工具分析

    隨著前端技術的不斷發展和應用程式的日益復雜,前端自動化測驗也在不斷演進。隨著 Web 應用程式變得越來越復雜,自動化測驗的需求也越來越高。如今,自動化測驗已經成為 Web 應用程式開發程序中不可或缺的一部分,它們可以幫助開發人員更快地發現和修復錯誤,提高應用程式的性能和可靠性。 ......

    uj5u.com 2023-04-20 07:43:16 more
  • CANN開發實踐:4個DVPP記憶體問題的典型案例解讀

    摘要:由于DVPP媒體資料處理功能對存放輸入、輸出資料的記憶體有更高的要求(例如,記憶體首地址128位元組對齊),因此需呼叫專用的記憶體申請介面,那么本期就分享幾個關于DVPP記憶體問題的典型案例,并給出原因分析及解決方法。 本文分享自華為云社區《FAQ_DVPP記憶體問題案例》,作者:昇騰CANN。 DVPP ......

    uj5u.com 2023-04-20 07:43:03 more
  • msf學習

    msf學習 以kali自帶的msf為例 一、msf核心模塊與功能 msf模塊都放在/usr/share/metasploit-framework/modules目錄下 1、auxiliary 輔助模塊,輔助滲透(埠掃描、登錄密碼爆破、漏洞驗證等) 2、encoders 編碼器模塊,主要包含各種編碼 ......

    uj5u.com 2023-04-20 07:42:59 more
  • Halcon軟體安裝與界面簡介

    1. 下載Halcon17版本到到本地 2. 雙擊安裝包后 3. 步驟如下 1.2 Halcon軟體安裝 界面分為四大塊 1. Halcon的五個助手 1) 影像采集助手:與相機連接,設定相機引數,采集影像 2) 標定助手:九點標定或是其它的標定,生成標定檔案及內參外參,可以將像素單位轉換為長度單位 ......

    uj5u.com 2023-04-20 07:42:17 more
  • 在MacOS下使用Unity3D開發游戲

    第一次發博客,先發一下我的游戲開發環境吧。 去年2月份買了一臺MacBookPro2021 M1pro(以下簡稱mbp),這一年來一直在用mbp開發游戲。我大致分享一下我的開發工具以及使用體驗。 1、Unity 官網鏈接: https://unity.cn/releases 我一般使用的Apple ......

    uj5u.com 2023-04-20 07:40:19 more