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