基于注意力的轻量级工业产品缺陷检测网络

Lightweight Industrial Products Defect Detection Network Based on Attention

工业领域的表面缺陷检测对提高工业产品质量、维护生产安全具有重要意义。因工业产品表面缺陷复杂多样、形状各异、缺陷检测场景和硬件配置不同,对工业产品的表面缺陷检测提出更高要求。基于图像的工业产品表面缺陷检测方法难以兼顾实时性和准确性的要求。为满足工业产品缺陷检测快速准确的需求,提出一种轻量级的缺陷检测网络。该网络由主干网络、多尺度特征聚合网络、残差增强网络和注意力增强网络4部分组成。其中,主干网络将通道注意力层和坐标注意力层嵌入到特征提取部分,以获取丰富的表面缺陷特征信息,多尺度特征聚合网络则融合深层语义和浅层语义特征信息,残差增强网络关注空间信息,注意力增强网络利用全局特征与局部特征的信息交互,在满足低硬件配置的同时增强模型对复杂多样缺陷的检测性能。实验结果表明,该网络在NRSD-MN、NEU-DET和PCBData等公开数据集上的精准度、召回率、F1值、mAP@0.5和GFLOPS这5项指标上优于YOLOv3-tiny、YOLOv5s、YOLOv7-tiny等同参数量级算法,能有效兼顾工业产品表面缺陷检测场景下实时性和准确性的要求。

The detection of surface defects in industry is of great significance in improving the quality of industrial products and maintaining production safety.As surface defects are complex,diverse,and of different shapes,higher requirements are put forward for surface defect detection of industrial products in different defect detection scenarios and hardware configurations.The image-based surface defect detection method for industrial products cannot easily balance the requirements of real-time and accuracy.Thus,a lightweight defect detection network is proposed to meet the speed and accuracy requirements of industrial product defect detection.The proposed network consists of four parts:backbone,multi-scale feature aggregation,residual enhancement,and attention enhancement networks.Among them,the backbone network embeds the channel and coordinate attention layers into the feature extraction section to obtain rich surface defect feature information.The multi-scale feature aggregation network integrates deep and shallow semantic feature information.The residual enhancement network pays attention to spatial information,and the attention enhancement network utilizes information interaction between global and local features.The model detection performance for complex and diverse defects has to be enhanced while satisfying simple hardware configurations.The experimental results show that the network performs well on publicly available datasets such as NRSD-MN,NEU-DET,and PCBData,with respect to precision,recall,F1 value,and mean Average Precision(mAP)@0.5 values.Compared to algorithms such as YOLOv3-tiny,YOLOv5s,and YOLOv7-tiny,it can effectively balance the real-time and accuracy requirements of industrial product surface defect detection scenarios with respect to the five indicators of GFLOPS.