高级检索

重新思考工业微小缺陷检测中的掩模降采样和回归损失函数

Rethinking Mask Down Sampling and Regression Loss Function in Industrial Tiny Defect Detection

  • 摘要: 针对PCB在生产过程中会出现短路、漏孔等缺陷,已有的损失函数和降采样算法存在定位不准确等导致缺陷检测收敛缓慢、检测结果不准确的问题,提出了一种基于掩模降采样和微小缺陷交并比损失的一阶段缺陷检测器.首先,提出基于面积损失的微小缺陷交并比损失函数,以进行更精确的微小缺陷回归定位;其次,提出基于动态掩模的降采样算法,以利于检测器在缩小参数矩阵尺寸过程中自动筛选重要特征、次要特征和噪声特征,提升缺陷检测器的特征提取能力.实验结果表明,提出的缺陷检测器在北京大学PCB缺陷数据集和Deep PCB数据集上可分别达到98.50%的mAP和99.02%的mAP,优于对比算法;提出的降采样算法使YOLOv5等检测器的mAP在北京大学PCB数据集中提升了1.6个百分点,提出的损失函数有利于YOLOv3和YOLOv4等检测器提升其在2个数据集上的检测准确率,展现出良好的鲁棒性.

     

    Abstract: Aiming at the PCB in the production process will appear short circuit, missing holes and other defects, the existing loss function and down sampling algorithms have inaccurate localization and so on leading to the slow convergence of defect detection, inaccurate detection results, a mask down sampling and tiny defects based on the intersection of the ratio of the loss of the one-stage defect detector is proposed. First, an area loss-based loss function is proposed for more accurate regression localization of tiny defects. Second, a dynamic mask-based down sampling algorithm is proposed to facilitate the automatic screening of important, secondary and noisy features in the process of reducing the size of the parameter matrix, so as to improve the feature extraction capability of the defect detector. The experimental results show that the proposed defect detector can achieve 98.50% mAP and 99.02% mAP on the Peking University PCB defect dataset and Deep PCB dataset, respectively, which is better than the comparison algorithms; the proposed downsampling algorithm improves the mAP of the detectors such as YOLOv5 in the PCB dataset of Peking University by 1.6 percentage points. The proposed loss function facilitates the detectors such as YOLOv3 and YOLOv4 to improve their detection accuracies on the two datasets, showing good robustness.

     

/

返回文章
返回