高级检索

    YOLOv5算法在钨棒表面缺陷机器视觉检测中的应用

    Application of YOLOv5 algorithm in machine vision detection of surface defects on tungsten rods

    • 摘要: 表面缺陷检测是控制钨棒质量的重要环节。由于钨棒检测图像存在背景复杂、噪声干扰多、对比度低、缺陷分辨率低、尺度跨度广等问题,现有YOLOv5算法的检测准确率难以满足工业现场要求。提出一种基于YOLOv5算法的改进算法,首先重构网络模型的检测头,利用更多尺度的特征图进行结果预测;其次,在特征提取网络中添加注意力机制,增强算法对于重点特征的提取能力,同时采用加权双向特征金字塔网络结构实现多尺度特征融合,减少特征融合过程中的冗余,提高多尺度目标特征的融合度。试验结果表明,基于改进YOLOv5的钨棒表面缺陷检测算法更适应于具有小尺寸大曲度特点的金属表面缺陷检测,能够显著提高此类检测的准确率和精度。

       

      Abstract: Surface defect detection is an important step in controlling the quality of tungsten rods. Due to the complex background, high noise interference, low contrast, low defect resolution, and wide scale span of tungsten rod images, the detection accuracy of the existing YOLOv5 algorithm is difficult to meet the requirements of industrial sites. This paper proposed an improved algorithm based on YOLOv5, which first reconstructed the detection head of the network model and used more scale feature maps for result prediction; Secondly, an attention mechanism was added to the feature extraction network to enhance the algorithm's ability to extract key features. At the same time, a weighted bidirectional feature pyramid network structure was adopted to achieve multi-scale feature fusion, reducing redundancy in the feature fusion process and improving the fusion degree of multi-scale target features. The experimental results showed that the tungsten rod surface defect detection algorithm based on improved YOLOv5 was more suitable for metal surface defect detection problems with small size and large curvature characteristics, and could significantly improve the accuracy and precision of such detection.

       

    /

    返回文章
    返回