基于增强可变形卷积的带钢表面缺陷检测网络

Strip steel surface defect detection network based on enhanced deformable convolution

  • 摘要: 提出了一种面向带钢表面缺陷检测的新型网络架构,设计了基于协同注意力的增强可变形卷积模块,并将其作为插件集成至骨干网络;利用协同注意力机制,自适应调整卷积核形状,有效捕捉带钢表面不规则缺陷,显著提升骨干网络的特征提取能力。在NEU‑DET数据集上的实验结果表明,该方法在平均精度(mAP)上达到81.6%。

     

    Abstract: A novel network architecture for strip steel surface defect detection was proposed, which designed an enhanced deformable convolution module based on co‑attention mechanism and integrated it into the backbone network as a plugin. By leveraging the co‑attention mechanism, the shape of the convolution kernel was adaptively adjusted, which effectively captured irregular defects on the strip steel surface, and significantly improved the feature extraction capability of the backbone network. Experimental results on the NEU‑DET dataset demonstrated that the proposed method achieved an average precision (mAP) of 81.6%.

     

/

返回文章
返回