基于重建的工业图像异常检测通常假设模型能很好重建正常区域，而不能很好重建异常区域。但由于深度神经网络存在过度泛化问题，使得异常区域也能被较好重建，导致异常区域漏检。为解决上述问题，本文提出一种基于多记忆增强模块及图像轮廓重建的工业表面异常检测网络(Industrial Surface Anomaly Detection Based on Reconstruction with Multiple Memory Enhancement Modules and Image Edge，MMAERec)。具体来说，在带有跳跃连接的U-Net类型去噪自编码器上引入多记忆增强模块和图像轮廓提取模块。多记忆增强模块得到的记忆特征有利于很好地重建正常区域；而提取到的图像轮廓特征则有利于很好重建图像轮廓。将这两种不同特征的融合经注意力机制处理并用于重建能很好地提高重建图像质量。所提方法可以强制网络学习正常的低频和高频信息，防止模型直接复制异常区域，有效缓解过度泛化问题。在MVTec AD和BTAD两个工业数据集上的实验结果也展现了所提方法良好的检测和定位性能。
暂无评论