YOLOV4 windows10系统训练自己的图片数据集(robomaster官方数据集)
首先,当然是要配置好环境啦,这里相信屏幕前的你已经配置好了,如果没有配置好也不用着急。可以看一下本博主另一篇保姆教程,请移步:https://blog.csdn.net/weixin_39954922/article/details/105785460,,,,啊哈哈哈哈 一、配置好之后,最重要的就是数据集,因为博主报名参加了RM比赛,所以就用大疆给的官方数据集试验了一下,首先准备两个数据集,一个是图片数据集Image,一个是每个图像内标记好目标后生成的xml文件Annotation。这里我只用了两百张图片:如下图: 二、把图片放在darknet.exe同级目录下的data文件夹,在data内新建文件夹专门用来存放图片,这里我建立的是robomaster文件夹。 三、需要准备两种txt文件,一种是train.txt,这个txt文件要放在data文件夹下,且内容为存储了每张图片相对于darknet.exe的路径,笔者的文件如下: 另一种TXT文件是每张图片都对应有一个,是通过xml文件得来的,每个TXT文件里面都是纯数字如图一所示,记录了标记目标的信息。这个TXT的命名与对应图片的命名一样,且训练时要放置在与image相同目录下,如图2所示: 图一 图二 四、在data文件夹下建立.names和.data文件。笔者建立的是robomaster.names和robomaster.data文件,在robomaster.names文件里面写上要检测的目标的名称,每个名称占用一行,如图三所示,这里检测目标有五个 图三 robomaster.data文件的内容包括总共有多少类对象,train.txt的文件路径,备份路径等,如图四所示。classes要改成对应的数字,如果你想检测的目标有6个,就改成6。 图四 五、建立训练所需的.cfg文件,在cfg文件夹下创建,笔者直接复制的原有的yolov4.cfg,并改名为yolov4-robomaster.cfg。 六、更改cfg文件内的参数。非常重要!!! 6.1更改batch为batch=64 6.2更改subdivisions为subdivisions=16 6.3更改max_batches为classes数*2000,但是不要少于4000,例如笔者这里有classes=5,那么max_batch...