基于YoloV5改进模型的汽车保险丝目标检测方法与流程

本发明涉及汽车保险丝检测，具体为一种基于yolov5改进模型的汽车保险丝目标检测方法。

背景技术：

1、汽车保险丝是一种用于保护汽车内部电路的设备，当电路电流超过保险丝额定电流时，保险丝会自动熔断，起到电路保护的作用。汽车保险丝生产时，需要将保险丝安装于保险丝盒1内，但实际加工生产过程中，极易出现保险丝2安装错误等问题，若将安装错误的保险丝用于汽车内，则无法起到电路保护作用，甚至可能因保险丝无法正常熔断而引起严重事故。因此，需要对保险丝盒内的保险丝进行检测。

2、传统的保险丝目标检测主要基于图像处理技术实现，但保险丝盒内背景较为复杂，对保险丝位置、光线亮度以及相机拍摄角度等具有严格要求，适应性较差。另外，在图像处理过程中，常采用图像拼接等方式对目标进行拼接处理，然后再通过图像分割、模板匹配等方式获取图像中保险丝目标，图像拼接使图像中面积较小的保险丝与面积较大的保险丝拼接为一体，导致后续目标检测时，无法实现小目标保险丝的有效识别，严重影响了保险丝目标检测准确性。

技术实现思路

1、针对现有技术中存在的保险丝目标检测准确性差的技术问题，本发明提供了一种基于yolov5改进模型的汽车保险丝目标检测方法，其可实现图像中小目标保险丝的有效识别，可提高保险丝目标检测准确性。

2、为实现上述目的，本发明采用如下技术方案：

3、一种基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，该方法基于yolov5改进模型实现，具体步骤包括：

4、s1、使用工业相机采集待测汽车保险丝的原始图像；

5、s2、对所述原始图像中不同保险丝进行数据标注，获取标注图像；

6、s3、将所述标注图像划分为训练集、测试集；

7、s4、将所述训练集导入yolov5改进模型中进行训练，获取yolov5改进模型，训练步骤包括：

8、s41、在yolov5模型中添加深层特征提取模块，提取深层特征；

9、s42、将提取出的深层特征与浅层特征进行特征融合，获取融合图像；

10、s43、对所述融合图像进行解耦，获取检测目标，实现yolov5改进模型训练；

11、s5、基于建立的所述yolov5改进模型，对所述测试集进行目标检测，获取测试集的检测目标；

12、s6、将测试集的检测目标与实际目标进行对比，判断检测准确性，若测试集的检测目标与测试集的实际目标一致，则表明检测准确，反之，则表明检测不准确。

13、其进一步特征在于，

14、所述yolov5改进模型包括：特征提取和目标检测，其中，特征提取包括卷积模块、sppf模块、特征耦合，所述卷积模块用于将原始图像中的特征处理为深层特征；sppf模块用于对所述深层特征进行增强处理，

15、步骤s1中，所述原始图像为灰度图；

16、步骤s2中，使用图像标注工具(labelimg)对所述原始图像中不同保险丝进行标注，获取所述标注图像；

17、步骤s3中，将所述标注图像按比例7：3划分为训练集、测试集；

18、步骤s41中，所述深层特征提取模块包括所述卷积模块、sppf模块，依次采用卷积模块、sppf模块对所述标注图像进行分割、增强；

19、步骤s41中，深层特征为所述标注图像中的1/64尺度大小特征信息；

20、步骤s43中，所述解耦处理包括目标分类、目标回归，对所述融合图像进行解耦指：依次采用目标分类、目标回归对所述融合图像中不同层次的保险丝目标进行识别、定位，获取训练集的检测目标；

21、其中，不同层次的保险丝目标指图像中不同面积大小的保险丝目标。

22、采用本发明上述方法可以达到如下有益效果：本申请基于yolov5改进模型对汽车保险丝进行检测，在yolov5改进模型训练过程中，使用深层特征提取模块，提取标注图像中的深层特征信息，深层特征信息保留了图像中更加细化的图像特征信息，更加细化的图像特征信息主要用于表征面积较小的保险丝目标，从而防止了面积较小的保险丝目标未被有效识别而导致目标检测准确性降低的问题出现，因此，yolov5改进模型中，深层特征信息的提取，有利于提高保险丝目标检测准确性。

23、本申请yolov5改进模型中采用了特征融合方法，特征融合方法能够针对不同层次的目标信息进行分别处理，有利于识别和定位不同层次的目标信息，从而确保了原始图像中面积较小的保险丝被有效识别，提高了保险丝目标检测准确性。

技术特征：

1.一种基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，该方法基于yolov5改进模型实现，具体步骤包括：

2.根据权利要求1所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，步骤s1中，所述原始图像为灰度图。

3.根据权利要求1所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，步骤s2中，使用图像标注工具对所述原始图像中不同保险丝进行标注，获取所述标注图像。

4.根据权利要求1所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，步骤s3中，将所述标注图像按比例7：3划分为训练集、测试集。

5.根据权利要求1所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，所述yolov5改进模型包括：特征提取和目标检测，其中，特征提取包括卷积模块、sppf模块、特征耦合，所述卷积模块用于将原始图像中的特征处理为深层特征；sppf模块用于对所述深层特征进行增强处理。

6.根据权利要求5所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，步骤s41中，所述深层特征提取模块包括卷积模块、sppf模块，依次采用卷积模块、sppf模块对所述标注图像进行分割、增强。

7.根据权利要求6所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，步骤s41中，深层特征为所述标注图像中的1/64尺度大小特征信息。

8.根据权利要求1或7所述的基于yolov5改进模型的汽车保险丝目标检测方法，其特征在于，步骤s43中，所述解耦处理包括目标分类、目标回归，对所述融合图像进行解耦指：依次采用目标分类、目标回归对所述融合图像中不同层次的保险丝目标进行识别、定位，获取训练集的检测目标。

技术总结
本发明公开了一种基于YoloV5改进模型的汽车保险丝目标检测方法，其可实现图像中小目标保险丝有效识别，可提高目标检测准确性，该方法基于YoloV5改进模型实现，检测步骤包括：采集待测汽车保险丝原始图像，对原始图像中不同保险丝进行数据标注，将标注图像划分为训练集、测试集，将训练集导入YoloV5改进模型中进行训练，获取YoloV5改进模型，训练步骤包括：提取深层特征，将提取出的深层特征与浅层特征进行特征融合，对融合图像进行解耦，获取检测目标；基于YoloV5改进模型，对测试集进行目标检测，获取测试集的检测目标，若测试集检测目标与测试集实际目标一致，则表明检测准确，反之，则表明检测不准确。

技术研发人员：刘国营,陈鹏飞,席云斌,葛继
受保护的技术使用者：无锡图创智能科技有限公司
技术研发日：
技术公布日：2024/1/12