win10下darknet训练车辆数据集

最新推荐文章于 2023-12-13 11:49:28 发布

海清

最新推荐文章于 2023-12-13 11:49:28 发布

阅读量1.3k

点赞数

文章标签：深度学习 python 机器学习

本文链接：https://blog.csdn.net/qq_35622355/article/details/103886482

版权

win10下darknet训练车辆数据集的学习过程，部分内容来源网络分享，如书写涉及侵权，将立即删除

前提：
1.darknet框架编译成功
2.UA-DETRAC数据集

目的：
在darknet框架下训练DETRAC数据集，检测目标车辆

准备：
darknet.exe
DETRAC数据集中的DETRAC-train-data及DETRAC-Train-Annotations-XML包

过程：
1.数据集格式转换
将DETRAC的数据集转换为VOC格式
（该部分来源于CSDN搜索，链接https://blog.csdn.net/weixin_38106878/article/details/88684280和https://blog.csdn.net/liushangkun/article/details/89518995）
第一部分，把DETRAC-Train-Annotations-XML中的xml文件依次提取出每张图片的voc格式的xml，代码为DETRAC_xmlParser.py
第二部分：根据生成的XML文件，迁移相应的图片到目标目录中。代码为voc_data_migrate.py，图片和XML文件生成后，按照VOC数据格式的目录结构来制作该数据；
第三部分：利用代码ImageSets_Convert.py, 产生trainval.txt,test.txt,train.txt,val.txt文件；
2.网络结构的修改
需要修改三份文件，分别为：
cfg目录下的网络配置文件:**.cfg文件、.data文件；
data目录下的 .names文件；
具体的修改方法如下文所示；
yolov3.cfg文件的修改,共计三处需要修改，改完后在同一目录下另存为car.cfg：
由于我们只需要做一类目标的检测，因此，class数量为1, 需注意：模型训练的batch以及subdivision请依据自己硬件水平来调整；

cfg文件夹下的car.data修改为

classes= 1
train  = D:/program/UA-DETRAC/VOC2007/ImageSets/Main/train.txt
valid  = D:/program/UA-DETRAC/VOC2007/ImageSets/Main/test.txt
names = data/car.names
backup = backup

note:train和valid的路径需要写成绝对路径，写成相对路径会出错

cfg目录下新建一个car.names
文件内容是

car

模型的训练
此部分可参考darknet的官网
1.darknet官网上下载预训练模型-darknet53.conv.74
2.开始训练模型
在终端内输入指令
darknet.exe detector train cfg/car.data cfg/car.cfg darknet53.conv.74 -gpu 0,1

note：window系统下打开cmd输入darknet.exe —操作命令，linux系统下或者git和cygwin的界面下输入的是./darknet –
在这个底下有说明https://github.com/AlexeyAB/darknet#how-to-compile-on-windows-using-vcpkg

训练过程中遇到的报错
1.cannot load image
我遇到的这个问题是由于开始生成的文本中，只注明了图片的名字，但是没有写图片的路径，后来修改了生成的train和test文本中的内容就可以读取到图片了在这里插入图片描述

2.can’t open label file

遇到这个问题的原因是没使用官方的voc_label.py这个文件去生成合适darknet训练的数据格式，所以要将xml格式转换为txt格式。

还需要做的事情
1.GPU如何利用起来，如何修改makefile文件
2.darknet下命令行的使用
darknet.exe detector train cfg/car.data cfg/car.cfg darknet53.conv.74
darknet.exe detector test cfg/car.data cfg/car.cfg backup/car_2000.weights VOC2007/JPEGImages/MVI_20011__img00001.jpg
darknet.exe detector map cfg/car.data cfg/car.cfg backup/car_1000.weights

学习来源：
https://github.com/AlexeyAB/darknet#how-to-compile-on-windows-using-vcpkg
https://blog.csdn.net/liushangkun/article/details/89518995