融合判别式模型和生成式模型的缺陷识别方法及装置

本发明实施例公开了一种融合判别式模型和生成式模型的缺陷识别方法，所述方法包括：获取第一训练数据集，第一训练数据集包括至少一个训练样本，每一个训练样本包括样本图像以及与该样本图像对应的缺陷标注结果；获取生产参数集，生产参数集为外部输入的生产环节相关的要素参数的集合；通过预设的生成式模型，并融合生产参数集，对第一训练数据集进行生成处理，以获取第二训练数据集；基于第二训练数据集，对预设的判别式模型进行训练，得到训练完成的判别式模型；获取待检测图像，将待检测图像输入判别式模型，获取判别式模型的输出结果作为待检测图像的缺陷分类结果。采用本发明，可提高缺陷识别的准确率和识别效率。