yolov5 训练自己的数据集后,检测不到目标的解决办法

您所在的位置:网站首页 苹果图片搜索图片识别不出来 yolov5 训练自己的数据集后,检测不到目标的解决办法

yolov5 训练自己的数据集后,检测不到目标的解决办法

2024-07-08 09:26:51| 来源: 网络整理| 查看: 265

yolov5 训练自己的数据集后,检测不到目标的解决办法

1. 现象

yolov5 训练自己的数据集后,检测不到目标,表现有三个:一是在 “yolov5 Source Path\runs\train\exp8” (“yolov5 Source Path” 是 yolov5 源码路径,“exp8” 是训练后保存结果的文件夹,根据实际情况,可能会有所不同,主要是 “exp” 后面的数字不同) 中,“results.png” 中的前面几张曲线图上没有曲线;二是,验证集中的图片,没有识别结果,即 名字为 “val_batch*_pred.jpg” (* 是数字序号,根据实际情况,会有所不同) 的图片中,本来应该是显示验证集的识别结果,但是却没有识别出来,所以该有的识别框没有画出来,显示的还是验证集的原图片,同样的,此时,用 detect.py 去识别测试图片,一样识别不了,结果会显示 “no detections”。

该有的曲线却没有

第三个现象是,训练的过程中,每个 Epoch 后,会打印结果,然后我发现,所有的结果中,box_loss 和 obj_loss 下面的值都是 nan,而所有的 P 和 R 下面的值,都是 0(下面的图,我用的是正常训练时得到的图改的,异常训练的情况忘了截图)。我查了一下,说这是因为在训练的过程中,得到了 非法的数字,有很多原因,其中一个可能是 训练数据和它的标签有问题,至于是不是这个问题,后面说明。

训练异常示例

2. 解决办法

在网上查了一下,不同的方法都试了一下,结果都不行,后来实在没有办法了,在网上看到有人说可以降低 cuda 的版本,于是就试了一下,逐渐降低 torch 和 cuda 的版本,一个一个的低版本试,终于在 这个版本的时候,可以正常训练了,conda 下的 安装命令是: " pip install torch1.9.1+cu102 torchvision0.10.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html " 可以在官网: " https://pytorch.org/get-started/previous-versions/ " 上找到这个安装命令,其实是: " pip install torch1.9.1+cu102 torchvision0.10.1+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html " 。 但是我在安装时,报错了,说是要安装的 torchaudio 版本与要安装的 torch 版本冲突了,于是我就不安装 torchaudio 了,因为我用不上。安装好了后,再次训练,可以发现,输出的结果正常了,最后,该有的训练结果曲线也有了,验证集的图片也有了识别结果,然后用 detect.py 测试,测试图片也得到了正确的结果。

正常的训练结果

正常的训练结果曲线

3. 未解决的问题

可以看到,在上面的图中,还有一个警告: " train.py:330: FutureWarning: Non-finite norm encountered in torch.nn.utils.clip_grad_norm_; continuing anyway. Note that the default behavior will change in a future release to error out if a non-finite total norm is encountered. At that point, setting error_if_nonfinite=false will be required to retain the old behavior. torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm=10.0) # clip gradients " 。 它的意思是说,在计算中,得到了一个非法的数字,且以后的高版本中,这个警告不会打印出来了。前面说了,得到了 非法的数字,有很多原因,其中一个可能是 训练数据和它的标签有问题,但是,我觉得这可能不是我数据的原因(当然也有可能是),如果是数据的原因,每个 Epoch 应该都会报出这个警告,或者 每个 Epoch 都会得到 nan 的结果,但是,在这个可行的版本中,只在前面有两次报出了这个警告,后面一切正常,在高版本中,每次 Epoch 都会得到 nan 结果。

所以,这里报出这个警告以及在 高版本 torch 和 cuda 中训练时,每次都得到 nan 结果,如果有人知道其中原因,麻烦请告知,谢谢!



【本文地址】

公司简介

联系我们

今日新闻


点击排行

实验室常用的仪器、试剂和
说到实验室常用到的东西,主要就分为仪器、试剂和耗
不用再找了,全球10大实验
01、赛默飞世尔科技(热电)Thermo Fisher Scientif
三代水柜的量产巅峰T-72坦
作者:寞寒最近,西边闹腾挺大,本来小寞以为忙完这
通风柜跟实验室通风系统有
说到通风柜跟实验室通风,不少人都纠结二者到底是不
集消毒杀菌、烘干收纳为一
厨房是家里细菌较多的地方,潮湿的环境、没有完全密
实验室设备之全钢实验台如
全钢实验台是实验室家具中较为重要的家具之一,很多

推荐新闻


图片新闻

实验室药品柜的特性有哪些
实验室药品柜是实验室家具的重要组成部分之一,主要
小学科学实验中有哪些教学
计算机 计算器 一般 打孔器 打气筒 仪器车 显微镜
实验室各种仪器原理动图讲
1.紫外分光光谱UV分析原理:吸收紫外光能量,引起分
高中化学常见仪器及实验装
1、可加热仪器:2、计量仪器:(1)仪器A的名称:量
微生物操作主要设备和器具
今天盘点一下微生物操作主要设备和器具,别嫌我啰嗦
浅谈通风柜使用基本常识
 众所周知,通风柜功能中最主要的就是排气功能。在

专题文章

    CopyRight 2018-2019 实验室设备网 版权所有 win10的实时保护怎么永久关闭