高级检索

    一种基于改进YOLOv10算法的钢铁线材表面缺陷识别方法

    An identification method for surface defects of steel wires based on improved YOLOv10 algrithm

    • 摘要: 针对钢铁线材表面缺陷识别精度低、速度慢的问题,提出一种基于改进YOLOv10算法的钢铁线材表面缺陷识别方法。引入空间和通道重建卷积模块、多尺度分离增强注意力模块和交互式多头自注意力模块改进YOLOv10网络,以使其对于复杂表面缺陷数据的建模能力更精确,对于贴近或相互遮挡的小区域缺陷的识别能力更强,同时进一步降低特征的空间和通道冗余,轻量化网络结构,保持识别性能和计算开销之间的平衡。试验结果表明,改进后的YOLOv10缺陷识别网络的识别准确率和识别速度更具有竞争力。

       

      Abstract: Aiming at the problems of low accuracy and slow speed of surface defect identification of steel wire, a surface defect recognition method for steel wire based on improved YOLOv10 was proposed. A spatial and channel reconstruction convolution module, a multi-scale separation enhanced attention module and an interactive multi-head self-attention module were introduced to improve the YOLOv10 network, so that it had a more accurate modelling capability for complex surface defect data, and an obvious improvement in the recognition capability for the defects in a small area that were close to each other or mutually occluded, and at the same time, it further reduced the spatial and channel redundancy of the features, and lightened the structure of the network to maintain a balance between recognition performance and computational overhead. Experimental results showed that the improved YOLOv10 defect identification network achieved competitive recognition accuracy and recognition speed.

       

    /

    返回文章
    返回