面向高端装备产品小目标瑕疵图像检测方法及系统

本发明属于图像处理技术领域，本发明公开了面向高端装备产品小目标瑕疵图像检测方法及系统；其中，方法包括以下步骤：S1：建立并训练AttentionGAN模型；S2：建立并训练Pix2Pix模型；S3：瑕疵检测，具体包括：随机选择带瑕疵图像y以生成图像F(y)，逐像素比对带瑕疵图像y与修复后的无瑕疵图像F(y)，以获得瑕疵位置与形状的二值图像；本发明能够降低瑕疵图像合成的复杂性并解决检测未见过瑕疵的难题，相较于传统的朴素CycleGAN，能够获得更优的性能，且在效果上可与监督式UNET相媲美；另外，本发明通过引入注意力机制对图像中的瑕疵区域进行加权，能更精确地生成和检测未见过的瑕疵。