首先用自己的数据集训练的时候,必须首先训练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训练的全过程了,有什么错误或者我漏掉的可以私信我。