基于三重注意力机制的带钢表面显著性缺陷检测

Salient object detection for strip steel surface defects based on triple attention mechanism

  • 摘要: 带钢表面缺陷普遍存在低对比度、类内差异显著及类间相似性高的特点,对模型检测精度构成巨大挑战。为解决上述问题,提出一种基于三重注意力机制的显著性目标检测模型,旨在实现带钢表面缺陷检测。针对主干网络提取的粗缺陷特征,构建包含三个分支的特征细化模块,各分支从不同维度关注带钢表面缺陷的通道和空间特征。同时,为细化显著目标的边缘细节,构建一种局部注意力特征过滤模块,并将其嵌入到特征细化模块中,提高模型对缺陷区域的敏感性。实验结果表明,与现有方法相比,本文所提模型在显著性目标检测任务中展现显著优势,其平均绝对误差、 加权F测度等评价指标均优于同类方法,验证了所提方法的有效性和鲁棒性。

     

    Abstract: Surface defects on strip steel have characteristics such as low contrast, large intra⁃class differences, and inter⁃class similarity, which bring significant challenges to the accuracy of model detection. To address these challenges, a salient object detection model based on a triple⁃attention mechanism was proposed to achieve more accurate detection of surface defects on strip steel. A feature refinement module, consisting of three branches, was constructed for the coarse defect feature maps extracted from the model backbone network. Each branch was designed to focus on the channel and spatial features of surface defects on the strip steel from different perspectives. Additionally, to refine the edge details of salient targets, a local attention feature filter module was built and embedded into the feature refine module as a plugin, which enhances the model’s sensitivity in defect areas. The experimental results indicated that the proposed model had significant advantages over existing methods in salient object detection and outperforms similar methods in evaluation indicators such as mean absolute error and weighted F-measure, thereby verifying the effectiveness and robustness of the proposed method.

     

/

返回文章
返回