论文标题
正常参考注意力和缺陷特征感知网络用于表面缺陷检测
Normal Reference Attention and Defective Feature Perception Network for Surface Defect Detection
论文作者
论文摘要
视觉异常检测在工业自动产品质量检查的发展中起着重要作用。由于正常数据和异常数据的最大不平衡,对缺陷检测的无监督方法的注意力日益增长。尽管最近对现有基于重建的方法进行了广泛的研究,但由于均匀和非规范的表面纹理,建立了针对各种纹理表面缺陷检测的强大重建模型仍然是一项艰巨的任务。在本文中,我们提出了一种基于无监督重建的新型方法,称为正常参考注意力和有缺陷的特征感知网络(NDP-NET),以准确检查各种纹理缺陷。与大多数基于重建的方法不同,我们的NDP-NET首先采用编码模块,该模块提取表面纹理的多尺度区分特征,该模块具有通过所提出的人造缺陷和新颖的像素级缺陷感知损失的缺陷判别能力增强。随后,提出了一种新型的基于参考的注意模块(RBAM),以利用固定参考图像的正常特征来修复有缺陷的特征并限制缺陷的重建。接下来,将修复的功能馈送到解码模块中,以重建正常的纹理背景。最后,引入了新型的多量表缺陷分割模块(MSDSM)以进行精确的缺陷检测和分割。此外,还利用了两阶段的培训策略来增强检查性能。
Visual anomaly detection plays a significant role in the development of industrial automatic product quality inspection. As a result of the utmost imbalance in the amount of normal and abnormal data, growing attention has been given to unsupervised methods for defect detection. Although existing reconstruction-based methods have been widely studied recently, establishing a robust reconstruction model for various textured surface defect detection remains a challenging task due to homogeneous and nonregular surface textures. In this paper, we propose a novel unsupervised reconstruction-based method called the normal reference attention and defective feature perception network (NDP-Net) to accurately inspect a variety of textured defects. Unlike most reconstruction-based methods, our NDP-Net first employs an encoding module that extracts multi scale discriminative features of the surface textures, which is augmented with the defect discriminative ability by the proposed artificial defects and the novel pixel-level defect perception loss. Subsequently, a novel reference-based attention module (RBAM) is proposed to leverage the normal features of the fixed reference image to repair the defective features and restrain the reconstruction of the defects. Next, the repaired features are fed into a decoding module to reconstruct the normal textured background. Finally, the novel multi scale defect segmentation module (MSDSM) is introduced for precise defect detection and segmentation. In addition, a two-stage training strategy is utilized to enhance the inspection performance.