Windows10+Darknet+YOLOv3训练自己数据(GPU)——4.开始自己数据训练

本文详细介绍了在Windows10环境下使用Darknet和YOLOv3训练自定义数据集的步骤,包括创建VOCdevkit目录,转换XML数据,修改配置文件,调整训练参数,以及解决内存不足问题。
摘要由CSDN通过智能技术生成

4.开始自己数据训练

在阅读本篇前请先参考:

Windows10+Darknet+YOLOv3训练自己数据——1.数据集准备
Windows10+Darknet+YOLOv3训练自己数据——2.CUDA+cuDNN+Darknet框架配置
Windows10+Darknet+YOLOv3训练自己数据——3.已有模型测试

本篇主要参考该博文。我们在第一篇文章中已经准备好了JPEGImages和Annotations文件,下面准备模型训练需要的其他文件。

(1)在darknet的工程目录下,darknet\build\darknet\x64\data\下,新建立一个VOCdevkit目录,把VOC2007的目录放到里面。

在这里插入图片描述
VOC2007下面有这些文件,Annotations里面是标注过的xml文件,PEGImages里面是对应的jpg图像文件。

在这里插入图片描述
ImageSets下面有一个Main目录,最开始已经删空,下面用python创建Main目录里面的train.txt,val.txt,test.txt和trainval.txt这四个文件。在VOC2007目录下创建一个python代码,我这里命名xml2txt.py,具体代码如下:

import os
import random
 
trainval_percent = 0.1
train_percent = 0.9
xmlfilepath = 'Annotations'
txtsavepath = 'ImageSets\Main'
total_xml = os.listdir(xmlfilepath)
 
num = len(total_xml)
list = range(num)
tv 
  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值