流水化产品数据集增强方法、系统、设备及存储介质
摘要:
本发明提供一种流水化产品数据集增强方法、系统、设备及存储介质,属于产品缺陷检测技术领域。流水化产品数据集增强方法包括:获取多张待训练的工业产品图像,制作工业产品图像数据集;基于工业产品图像数据集训练预设的检测模型,得到工业检测模型;根据工业检测模型构建并训练生成对抗网络模型,获得生成对抗网络模型的生成器模型和判别器模型;使用生成器模型构建预备数据集,并基于预备数据集和工业产品图像数据集训练判别器模型,若判别器模型符合数据增强条件,在工业产品图像数据集中加入预备数据集。解决了当前部分工业产品检测训练集过少、难以获取、质量不高的问题。
0/0