融合注意力机制和感知损失的缺陷图像生成方法和系统

本发明公开了一种融合注意力机制和感知损失的缺陷图像生成方法，属于屏幕检测技术领域。方法包括：S1：获取真实图像；S2：以WGAN‑GP模型作为基础构建SWG‑VGG模型并进行训练，SWG‑VGG模型包括生成器、判别器和VGG‑19网络；VGG‑19网络用于训练时提取生成图像与真实图像的高层特征来计算这些特征之间的差异，最后将不同层的特征差异进行加权求和得到最终的感知损失并反馈给生成器；S3：将真实图像输入训练好的SWG‑VGG模型中得到生成图像。SWG‑VGG模型生成的数据集在mAP(50)上达到0.976，提升10.5%；生成图像的PSNR、SSIM、RMSE和CS值也表现出较高的图像质量。