一种基于改进YOLOv5算法的端塞焊焊缝图像缺陷检测方法

本申请属于缺陷检测技术领域，旨在解决人工检测存在的经验方面及长时间检测劳累等的问题，公开了一种基于改进YOLOv5算法的端塞焊焊缝图像缺陷检测方法，将YOLOv5模型中原始的CIOU Loss损失函数替换成EIOU Loss损失函数，添加SimAM注意力机制模块进入到原始的YOLOv5模型，使用改进后的YOLOv5模型进行训练得到训练的模型结果，运行训练好的改进YOLOv5模型对端塞焊焊缝图像进行测试，完成端塞焊焊缝图像的缺陷检测。本申请通过使用改进后的YOLOv5模型，对测试图片进行测试，缺陷检测结果的准确度较高、运行速度较快，能够实现端塞焊焊缝图像的缺陷检测要求。