Yolov5训练模型没有框

您所在的位置:网站首页 word2003图片不显示只有框 Yolov5训练模型没有框

Yolov5训练模型没有框

2024-03-09 07:20| 来源: 网络整理| 查看: 265

问题描述

今天在train.py上跑了自己的数据和权重,拿到detect.py上运行的时候可以正常跑,但是始终没有anchor,并且提示no detections。

 然后又拿官方的权重试了下发现可以正常anchor,所以怀疑是自己权重文件的问题,决定重新跑。

尝试方法

YOLO训练得到权重后无法检测detect目标_yolov5训练后识别不出来_小盆宇~的博客-CSDN博客参考了这个帖子。首先我用别人的权重能正常出结果那么一定不是我opencv的问题;遂试了在detect.py里面增加:cudnn.benchmark = True,提示cudnn没定义,我也搞不懂这是个啥玩意,于是放弃;我原本的epoch是30,增大到了300次,但是第187次的时候Early Stopping终止了程序,又试还是不行。还有一位大佬说把img_view打开,试过也没解决问题,而且在图片量较大时不建议打卡,会很卡。

解决方案

网上的方法在我这里都不管用,于是乎我就自己看我的代码,希望能发现问题的端倪,果不其然,问题在class这里,因为我之前拿别的数据集测,只用识别class=7这一类,所以把默认值也设成了7,而我现在跑的这个数据集只有0这一类,所以你说设个7他能出框就见鬼了。写在这里自己记录下,也提醒下小白,改参数的时候尽量在命令里改,不要动默认值。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3