arXiv:2601.11402cs.CV2026-01被引 2

针对印刷电路板微小缺陷检测难题,提出高效实时检测框架SME-YOLO。

SME-YOLO: A Real-Time Detector for Tiny Defect Detection on PCB Surfaces

  • 引入改进的损失函数与多尺度上采样模块,增强对微小缺陷的定位与细节还原能力。
  • 在PKU-PCB数据集上,mAP提升2.2%,精度提高4%,达到当前最优水平。
  • 适合工业质检场景,尤其适用于高纹理相似性下的微小缺陷识别。

印刷电路板(PCB)表面缺陷直接影响产品可靠性和安全性。然而,由于缺陷尺寸微小、纹理相似度高且尺度分布不均,实现高精度检测极具挑战。为此,本文提出基于YOLOv11n的新型框架SME-YOLO(Small-target Multi-scale Enhanced YOLO)。首先,采用归一化沃瑟斯坦距离损失(NWDLoss),有效缓解交并比(IoU)对微小目标位置偏差的敏感性;其次,用高效上采样卷积块(EUCB)替代原始上采样模块,通过多尺度卷积逐步恢复空间分辨率,增强边缘与纹理细节保留;最后,提出多尺度聚焦注意力模块(MSFA),针对PCB缺陷的空间分布特性,自适应强化关键尺度区间内的感知能力,实现局部细粒度特征与全局上下文信息的有效融合。在PKU-PCB数据集上的实验结果表明,SME-YOLO性能达当前最优水平,相较基线YOLOv11n,mAP提升2.2%,精度提高4%。

原文摘要 · Abstract (English)

Surface defects on Printed Circuit Boards (PCBs) directly compromise product reliability and safety. However, achieving high-precision detection is challenging because PCB defects are typically characterized by tiny sizes, high texture similarity, and uneven scale distributions. To address these challenges, this paper proposes a novel framework based on YOLOv11n, named SME-YOLO (Small-target Multi-scale Enhanced YOLO). First, we employ the Normalized Wasserstein Distance Loss (NWDLoss). This metric effectively mitigates the sensitivity of Intersection over Union (IoU) to positional deviations in tiny objects. Second, the original upsampling module is replaced by the Efficient Upsampling Convolution Block (EUCB). By utilizing multi-scale convolutions, the EUCB gradually recovers spatial resolution and enhances the preservation of edge and texture details for tiny defects. Finally, this paper proposes the Multi-Scale Focused Attention (MSFA) module. Tailored to the specific spatial distribution of PCB defects, this module adaptively strengthens perception within key scale intervals, achieving efficient fusion of local fine-grained features and global context information. Experimental results on the PKU-PCB dataset demonstrate that SME-YOLO achieves state-of-the-art performance. Specifically, compared to the baseline YOLOv11n, SME-YOLO improves mAP by 2.2% and Precision by 4%, validating the effectiveness of the proposed method.

缺陷检测YOLO工业视觉小目标

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。