一种生成与判别协同优化的工业视觉检测系统和方法

本发明提供了一种生成与判别协同优化的工业视觉检测系统和方法，涉及计算机视觉与人工智能领域。包含统一编码器、重构解码器、条件生成解码器和判别头；统一编码器用于基于待检测图像提取通用视觉特征；重构解码器用于得到不包含缺陷的重构图像；条件生成解码器用于生成包含指定缺陷的缺陷图像；判别头用于对重构图像进行识别，基于识别结果输出置信度；以及基于条件生成解码器生成的缺陷图像进行训练，生成缺陷类别和类别概率；协同优化模块用于构建协同损失函数、任务损失函数和总损失函数，基于总损失函数产生的梯度反向更新判别头参数、生成解码器参数和统一编码器参数。实现了生成模型与判别模型的共同进化，能有效提升工业视觉检测的性能。