Yolov5训练模型没有框

问题描述

今天在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他能出框就见鬼了。写在这里自己记录下，也提醒下小白，改参数的时候尽量在命令里改，不要动默认值。