Yolov5训练模型没有框

一、问题描述：

在用yolov5训练自己的图片，先运行train.py文件再运行detect.py文件后，发现测试集中的图片没有出现锚框。我参考的配置教程是：
https://blog.csdn.net/m0_53392188/article/details/119334634?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163694460616780271595311%2522%252C%2522scm%2522%253A%252220140713.130102334…%2522%257D&request_id=163694460616780271595311&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_alltop_positive~default-1-119334634.first_rank_v2_pc_rank_v29&utm_term=yolov5&spm=1018.2226.3001.4187

二、尝试的办法
1、更换配置环境：本来安装的是cuda11.3和cudnn8.2.1，后来换成cuda10.2和cudnn7.6.5。没有作用，还是没有出现锚框。

2、在detect.py里面增加：cudnn.benchmark = True，没用。
参考网站：https://blog.csdn.net/hh571050143/article/details/119984840?utm_term=yolov5%E6%B5%8B%E8%AF%95%E5%9B%BE%E7%89%87%E6%B2%A1%E6%9C%89%E6%A0%87%E7%AD%BE&utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2_allsobaiduweb~default-1-119984840&spm=3001.4430

3、自适应锚框：由于我要检测的目标实在太小，我怕里面的锚框设置有问题，于是参考网上大神修改autoanchor.py文件里面的K聚类自适应锚框的路径，搞了个test.py文件单独运行kmean_anchors函数，将获得的锚框参数置换到mydata.yaml文件相应位置中，运行完train.py后再运行detect.py，但是任然没用效果。（参考网址忘记储存了）

4、增加训练集：本来我的训练集是280张图片，后面通过翻转、旋转、提高亮度和对比度等方法，将训练集扩充到1000张图片。但是仍然没有出现锚框。
【图像分类–图片数据扩充】图片数据不够怎么凑：https://blog.csdn.net/u013421629/article/details/101314178?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163853284316780269845691%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=163853284316780269845691&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_allfirst_rank_ecpm_v1~rank_v31_ecpm-1-101314178.first_rank_v2_pc_rank_v29&utm_term=%E5%9B%BE%E7%89%87%E6%95%B0%E6%8D%AE%E6%89%A9%E5%85%85%E7%9A%84%E6%96%B9%E6%B3%95&spm=1018.2226.3001.4187

5、降低置信度：我将detect.py中的置信度值和IOU值都降低，再运行后，终于发现了问题。
https://blog.csdn.net/Nire_Yeyu/article/details/105403220

我发现我检测出来的标签是错的，根本不是我要的标签。马上反应过来是某个路径错了。

6、更换权重路线：我发现原来我测试的权重有问题，还是用的原来的王者荣耀的那个权重。因此把他改过来，换成我自己训练出来的best.pt后，才终于成功了。

三、最终解决办法
如上所述，我最终的解决办法是更换了权重，其实这都不是算法的错误，是自己粗心大意了。因此，很赞同有一篇文章说的那样。【目标检测 YOLOv5 - 如何提高模型的指标，提高精确率，召回率，mAP等：https://blog.csdn.net/flyfish1986/article/details/120704968?ops_request_misc=&request_id=&biz_id=102&utm_term=yolov5%20%E7%9A%84confusion%20matrix%E6%84%8F%E4%B9%89&utm_medium=distribute.pc_search_result.none-task-blog-2_allsobaiduweb~default-5-120704968.first_rank_v2_pc_rank_v29&spm=1018.2226.3001.4187】
我们还是得先保证自己用这个程序得时候没用错误，位置引用都对了，再调一些参数就好了。

Yolov5训练模型没有框相关推荐

（20）目标检测算法之YOLOv5计算预选框、详解anchor计算
目标检测算法之YOLOv5计算预选框.详解anchor计算单节段目标检测算法中:预选框的设定直接影响最终的检测精度众所周知,yolov5中采用自适应调整预选框anchor的大小,但万事开头难,配置 ...
yolov5训练模型
目录划分数据集生成数据集路径txt文件(xml转txt) 编写配置文件 data cfg 训练模型训练可视化报错运行在远程服务器检测目标(查看效果) 划分数据集 split_train_v ...
Yolov3训练模型没有框（理论上一定有用的解决方案）
问题描述不知道有多少人遇到过和我一样的情况,这个问题困扰了我很长时间,这里特别做一下笔记. 本人在Windows10环境下,使用Pycharm进行模型训练.训练的时候一切正常,loss值也有明显下降 ...
Yolo实用指南（step by step）之三yolov5训练自己的数据集
关于yolov5训练自己的数据集的文章网上已经有一大把,自己之前也摸索了数据标注有一阵子,可一直迟迟未动手,感觉对原理不熟悉,很多东西浮光掠影有些肤浅,思来想去还是花了整整一天实践一下吧,毕竟这是基础 ...
Debug Tensorflow: yolo模型检测框位置正确但类别不正确
平台 tensorflow yolov5 现象标注框没问题,但类别不正确解决首先确保后处理过程没有问题,nms无误由于yolo损失函数包含三部分,参考公众号YueTan.我的猜测是三部分可能没 ...
神经网络：训练模型+转化为k210上跑的kmodel
目录前言第一种第二种第三种第四种结语前言:之前用YOLOv5训练了一个口罩检测模型,但是想让模型在k210上运行,涉及到模型的转换,下面我就我个人转化历程来简单记录一下,我的艰辛转换历程 ...
Yolov5系列(3)-loss解析
Abstract 在yolov5中,loss在训练中起到了决定性的作用,同时,yolov5的loss又与大部分传统的方法不同,它是基于网格的.在网格上生成相应的anchor框和其对应的cls以及con ...
Yolov5算法解读
yolov5于2020年由glenn-jocher首次提出,直至今日yolov5仍然在不断进行升级迭代. Yolov5有YOLOv5s.YOLOv5m.YOLOv5l.YOLOv5x四个版本.文件中, ...
yolov5 anchors设置详解
yolov5中增加了自适应锚定框(Auto Learning Bounding Box Anchors),而其他yolo系列是没有的. 一.默认锚定框 Yolov5 中默认保存了一些针对 coco数据 ...

Yolov5训练模型没有框

Yolov5训练模型没有框相关推荐

最新文章

热门文章