基于软掩膜对比损失的图像异常检测方法
摘要:
本发明公开了一种基于软掩膜对比损失的图像异常检测方法,主要解决现有技术异常检测种类少,检测精度差的问题。其方案是:利用超像素分割算法与高斯噪声对MVTec数据集进行处理,得到合成图像训练数据集;搭建由重建自编码器和掩膜自编码器组成的异常检测模型,并使用L2误差损失作为掩膜自编码器的损失函数,使用L2误差损失与软掩膜对比损失作为重建自编码器的损失函数;利用合成图像训练数据集对其掩膜编码器和重建自编码器进行交替训练,直至各自的损失函数收敛;将测试图像输入到训练好的图像异常检测模型,得到图像异常检测与定位的结果。本发明异常识别种类多,检测精度高,可用于工业品瑕疵检测、医疗辅助诊断、视频违规行为检测。
公开/授权文献
0/0