YOLOX训练自己的数据集

前面的准备工作我已经在这篇博文写过了
YoloX训练VOC数据集(自用)自己运行踩坑,在windows系统运行
下面就是做数据集准备和修改相关代码了

1.数据集准备

使用voc的数据集目录结构如下

VOC2007
--Annotations
	--000000.xml
	--000001xml
--ImageSets
	--main
		--test.txt
		--trainval.txt
		--val.txt
		--train.txt
--JPEGImages
	--000000.jpg
	--000001.jpg

VOC格式的分布如上图所示

JPEGImages:存放数据集图片

Annotations文件夹:存放与图片对应的xml文件

ImageSets/Main文件夹:存放train.txt和val.txt

2.修改训练配置参数和类别

  1. 根据自己数据集的类别,修改yolox/data/datasets/voc_classes.py中的标签信息
    在这里插入图片描述

  2. 修改exps/example/yolox_voc/yolox_voc_s.py中的self.num_classes
    在这里插入图片描述

  3. 修改yolox/exp/yolox_base.py中的self.num_classes
    在这里插入图片描述

  4. 修改训练集数据目录,修改exps/example/yolox_voc/yolox_voc_s.py中的VOCDection
    因为这里用的是自己的数据集,所以需要修改
    在这里插入图片描述
    data_dir:修改为数据集文件夹的绝对路径
    images_sets:修改为train
    max_labels:表示图片最多的目标数量,根据数据集的具体情况进行修改
    修改yolox/data/datasets/voc.py中,VOCDection函数(125行)中的读取txt文件
    因为数据集里没有年份的信息,所以需要修改
    在这里插入图片描述
    加红的是原来的代码,

  5. 修改验证集信息,修改exps/example/yolox_voc/yolox_voc_s.py中的get_eval_loader函数
    在这里插入图片描述

  6. 其他需要修改的地方
    在yolox/data/datasets/init.py中添加调用的路径
    在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值