一种PCB缺陷图像生成方法

本申请涉及一种PCB缺陷图像生成方法，实现缺陷特征丰富度高，类间分布均衡的PCB缺陷图像生成，可用于增强AI检测模型。所述方法包括：采集生产现场的真实PCB缺陷图像，得到原始图像数据集；提取所述原始图像数据集的多模态特征，得到第一多模态数据集，所述多模态特征包括边缘特征、深度特征和文本描述；将所述原始图像数据集和所述第一多模态数据集输入到PCB缺陷图像生成模型进行第一模型训练，得到训练好的PCB缺陷图像生成模型；根据PCB缺陷的类别分布情况设置生成比例，生成缺陷生成数据集；提取所述缺陷生成数据集的多模态特征，得到第二多模态数据集；将所述第二多模态数据集输入到训练好的PCB缺陷图像生成模型，输出得到生成PCB缺陷图像。