YOLOV模型:基于YOLOX 注意力机制的魅力,训练自己的数据集

首先用自己的数据集训练的时候,必须首先训练yolox,然后将yolox训练好的模型当做yolov的预训练模型,然后训练yolov。作者的原话是: yolov should take the pretrain weights of yolox for we do not BP paramaters in yolox model。

1.训练yolox,如果你的数据集已经是yolov的数据集的格式,你可以使用以下代码将vid格式的xml转json。这里要注意一下代码中的图像的后缀,看是否是JPEG,我的图像是jpg,而代码中的后缀是JPEG,由于这个路径原因,我找了好长时间。代码还是比较简单的,大家一看就会了。

将多个文件夹或单文件夹内的xml文件转换为一个json标签(imagenet VID等视频目标检测数据集)简单易改,有注释_烤荔枝的大猫的博客-CSDN博客

代码复制的这个连接,大家可以直接点进去。 

2.修改yolox代码部分

yolox的运行是在train.py文件中

 这个修改的部分是看你选的哪一个模型,你如果选的是yoloxl_vid.py那就在这个模型中更改,记住这个路径是annotations的上一级。

 这个部分你之前训练过yolov你应该也改过 

D:\Aware_model\YOLOV-master\yolox\data\datasets\vid_classes.py

D:\Aware_model\YOLOV-master\yolox\data\datasets\vid.py

yolox错误部分,这个错误最好也是在训练之前改好,有时候报错不一样,但都是这个问题。

YOLOX复现BUG解决_caigou一号的博客-CSDN博客

 3.训练yolov

yolov数据怎么转.npy参考下面的博客

yolov模型数据集介绍,以及将数据集转.npy格式,可用于自建数据集,用yolov训练自己的数据集_一勺汤的博客-CSDN博客

这里选着的模型要对应,我yolox选的yolox-l,yolov选的也是yolov-l 

 验证集数据读取部分

 这是在yolov_l.py文件中,一般训练数据路径错误,你们可以单独吧报错的路径拿出来,写一个读取图片的几行代码看能否读取图片,如果不能就找个图片的路径和这个路径对比一下,看哪里错误,然后在修改路径。

 yolov要改的错误,最好把这个改了,有的还是报错不一样,但也是这个错误引起的

yolov 训练出现的时候验证出现错误 bug修复_一勺汤的博客-CSDN博客

 以上就是yolov训练的全过程了,有什么错误或者我漏掉的可以私信我。

  • 7
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值