yolov3 keshihua

by · 2020年07月25日 · 581 Words · ~2min reading time | Improve on


原文链接： yolov3 keshihua

darknet优化经验-AlexeyAB大神经验 - pprp - 博客园

8.用自己训练的权重作为预训练

有时候训练到一半突然终止了，这时候从头开始训练又很费时间，此时我们可以将自己之前保存的权重作为预训练权重。但是直接使用yolo-obj_last.weights会报错。需要做出如下转变。

#首先用第一行代码将yolo-obj_last.weights转化为olo-obj_last.conv.23
./darknet partial cfg/yolo-obj.cfg backup/yolo-obj_last.weights backup/yolo-obj_last.conv.23 23
#第二行将我们刚转化好的yolo-obj_last.conv.23作为预训练权重训练
./darknet detector train cfg/obj.data cfg/yolo-obj.cfg backup/yolo-obj_last.conv.23

重新计算你的数据集的anchor:(注意设置的时候计算问题)

darknet.exe detector calc_anchors data/obj.data -num_of_clusters 9 -width 416 -height 416

检查数据集通过https://github.com/AlexeyAB/Yolo_mark

————————————————
版权声明：本文为CSDN博主「CSzhangjinchao」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_44771532/article/details/105495755

diff --git a/Makefile b/Makefile
index 5635f8f..4748367 100644
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,7 @@
-GPU=0
-CUDNN=0
-CUDNN_HALF=0
-OPENCV=0
+GPU=1
+CUDNN=1
+CUDNN_HALF=1
+OPENCV=1
 AVX=0
 OPENMP=0
 LIBSO=0
diff --git a/cfg/voc.data b/cfg/voc.data
index eb6af25..6112953 100644
--- a/cfg/voc.data
+++ b/cfg/voc.data
@@ -1,6 +1,6 @@
-classes= 20
-train  = /home/pjreddie/data/voc/train.txt
-valid  = /home/pjreddie/data/voc/2007_test.txt
-names = data/voc.names
-backup = /home/pjreddie/backup/
+class=5
+train=/data/work/keshihua/trainval.txt
+valid=/data/work/keshihua/test.txt
+names=data/voc.names
+backup=backup

diff --git a/data/voc.names b/data/voc.names
index 8420ab3..4f38c7c 100644
--- a/data/voc.names
+++ b/data/voc.names
@@ -1,20 +1,6 @@
-aeroplane
-bicycle
-bird
-boat
-bottle
-bus
-car
-cat
-chair
-cow
-diningtable
-dog
-horse
-motorbike
-person
-pottedplant
-sheep
-sofa
-train
-tvmonitor
+DiaoChe
+TaDiao
+ShiGongJiXie
+DaoXianYiWu
+YanHuo

Tags:

`