高级检索

基于改进YOLOv5s模型的金属焊缝缺陷检测

Metal weld defect detection based on improved YOLOv5 model

  • 摘要: 为提高γ射线焊缝底片缺陷检测的可靠性,提出一种基于改进YOLOv5s的焊缝缺陷检测模型,实现了在复杂环境下对焊缝底片缺陷的高效准确检测. 针对原YOLOv5s模型中卷积网络存在的大量通道和冗余信息问题,首先,在主干网络的C3模块中融入了SCCONV网络模块,减少了冗余信息并提高模型检测性能;其次,考虑到焊缝缺陷的形态多样、大小不一以及对比度低等特点,引入了卷积注意力模块(CBAM和SE),以增强模型对感兴趣区域的关注度;最后,在边界框回归检测中,采用EIoU损失函数替代传统YOLOv5s中的CIoU损失函数,显著提升了模型的检测精度和鲁棒性. 结果表明,改进后的模型在精确度、召回率等指标上均较传统YOLOv5s算法有显著提升,具体表现为准确率提高4.2%、召回率提高3.2%、平均精度均值提高3.4%,进而验证了该方法在γ射线焊缝缺陷检测中的有效性.

     

    Abstract: To enhance the reliability of γ-ray weld image defect detection in radiographic films, this paper proposes an improved YOLOv5s-based defect detection model, achieving efficient and accurate identification of weld defects in complex environments. In response to excessive channels and redundant information in the convolutional network of the original YOLOv5s model. Firstly, we integrate the SCCONV module into the C3 block of the backbone network, reducing redundancy and boosting detection performance. Secondly, considering the characteristics of weld defects, such as morphological diversity, scale variation, and low contrast, convolutional attention modules (CBAM and SE) are introduced to enhance the attention of the model to the region of interest. Finally, the EIoU loss function replaces the traditional CIoU loss in bounding box regression detection, significantly improving detection accuracy and robustness. Experimental results demonstrate that the enhanced model outperforms the baseline YOLOv5s algorithm across key metrics. Specifically, the accuracy is increased by 4.2%, the recall rate is increased by 3.2%, and the mAP value is in-creased by 3.4%, which verifies the effectiveness of the method in the detection of γ-ray weld defects.

     

/

返回文章
返回