金属工件表面缺陷的结构–策略检测模型

    A Structure-Strategy Defects Detection Net for Metal Workpiece Surface

    • 摘要: 针对金属工件表面缺陷检测面临的缺陷形态复杂、尺度多样、边界模糊、标签质量不稳定、像素级标注成本高等问题,本文提出了一种端到端的结构-策略检测模型(Structure-Strategy Defect Detection Network, SSDDNet)。在结构层,构建多尺度上下文聚合模块,融合了动态卷积与多膨胀率信息,并引入边界增强模块来提供边界建模能力;在策略层,提出了空间标签不确定性建模,实现弱标注条件下的稳定训练。实验结果表明,SSDDNet 在公开数据集 KolektorSDD2 上无像素级标注条件下比当前主流模型MixSup的平均准确度提升1.1个百分点;在自建工业数据集 BatteryBase上,在弱标注条件下比 MixSup 模型的平均准确度提升了17.6个百分点,SSH指标提升了约15个百分点,这表明本文的模型具有良好的泛化性。本文的方法为金属工件表面缺陷检测提供了一种新思路。

       

      Abstract: An end-to-end Structure-Strategy Defect Detection Network (SSDDNet) is proposed in this research to solve the problems in surface defect detection of metallic workpieces, such as complex defect morphologies, diverse scales, ambiguous boundaries, unstable label quality, and the high cost of pixel-level annotation. A multi-scale contextual aggregation module is structurally established to fuse dynamic convolution and multi-dilation information, while a boundary enhancement module is introduced to strengthen the modeling of ambiguous boundaries. And a spatial label uncertainty modeling approach is strategically introduced to enable stable training under weakly annotated conditions. Experimental results show that: (1) SSDDNet achieves a 1.1% improvement in mean accuracy over the state-of-the-art MixSup model on the public KolektorSDD2 dataset without pixel-level annotations. (2) SSDDNet outperforms MixSup by 17.6 percentage points in mean accuracy and achieves approximately 15 percentage points improvement in SSH on the self-constructed industrial dataset BatteryBase, indicating the strong generalization capability of the proposed model. This research provides a novel approach for surface defect detection of metal workpieces.

       

    /

    返回文章
    返回