(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210237047.4 (22)申请日 2022.03.10 (71)申请人 深圳市腾盛精密装备股份有限公司 地址 518000 广东省深圳市龙华区大浪街 道新石社区新百丽工业园9号101- 502 (72)发明人 陈枫　郭江　黎春洁　江岱平　 卢国明　 (74)专利代理 机构 深圳中一联合知识产权代理 有限公司 4 4414 专利代理师 刘永康 (51)Int.Cl. G06K 9/62(2022.01) G06V 10/764(2022.01) G06V 10/774(2022.01) G06V 10/44(2022.01)G06V 10/82(2022.01) G01N 21/88(2006.01) (54)发明名称 一种数据增强方法、 装置、 终端设备及存储 介质 (57)摘要 本申请适用于工业视觉技术领域， 提供了一 种数据增强方法、 装置、 终端设备及存储介质。 本 申请实施例中获取缺陷样本图片， 根据上述缺陷 样本图片中的缺陷区域生 成缺陷图片， 并将上述 缺陷图片 存储于预设的缺陷样 本集中； 若上述缺 陷样本集中的缺陷图片满足预设条件， 则将上述 缺陷样本集中的缺陷图片随机覆盖到预设数量 的正常样 本图片上， 得到预设数量的目标样本图 片； 根据上述缺陷样本图片、 上述预设数量的正 常样本图片和上述预设数量的目标样本图片对 预设的模型进行训练， 从而提高了模 型识别结果 的准确率。 权利要求书2页 说明书10页 附图4页 CN 114638294 A 2022.06.17 CN 114638294 A 1.一种数据增强方法， 其特 征在于， 包括： 获取缺陷样本 图片， 根据所述缺陷样本 图片中的缺陷区域生成缺陷图片， 并将所述缺 陷图片存 储于预设的缺陷样本集中； 若所述缺陷样本集中的缺陷 图片满足预设条件， 则将所述缺陷样本集中的缺陷图片随 机覆盖到预设数量的正常样本图片上， 得到预设数量的目标样本图片； 根据所述缺陷样本图片、 所述预设数量的正常样本图片和所述预设数量的目标样本图 片对预设的模型进行训练。 2.如权利要求1所述的数据增强方法， 其特征在于， 在将所述缺陷图片存储于预设的缺 陷样本集中之后， 还 包括： 若所述缺陷样本集中的缺陷图片不满足预设条件， 则返回执行获取缺陷样本 图片， 直 至所述缺陷样本集中的缺陷图片满足预设条件。 3.如权利要求1所述的数据增强方法， 其特征在于， 所述根据所述缺陷样本图片中的缺 陷区域生成对应的缺陷图片， 包括： 确定所述 缺陷样本集中缺陷图片的目标缺陷类型； 根据所述目标缺陷类型从所述缺陷区域中确定目标区域， 所述目标区域的缺陷类型与 所述目标缺陷类型不同； 根据所述目标区域 生成所述 缺陷图片。 4.如权利要求3所述的数据增强方法， 其特征在于， 所述确定所述缺陷样本集中缺陷 图 片的目标缺陷类型， 包括： 获取所述缺陷样本集中缺陷图片的缺陷形状； 根据所述 缺陷形状确定所述 缺陷图片的目标缺陷类型。 5.如权利要求3所述的数据增强方法， 其特征在于， 所述根据所述目标区域生成所述缺 陷图片， 包括： 对所述目标区域做最小外 接矩形处 理， 得到矩形区域； 将边界区域的像素设为零， 所述边界区域为所述矩形区域内除所述目标区域之外的区 域； 根据所述矩形区域的各个 像素生成所述 缺陷图片。 6.如权利要求1至5任一项所述的数据增强方法， 其特征在于， 所述将所述缺陷样本集 中的缺陷图片随机覆盖 到预设数量的正常样本图片上， 包括： 从所述缺陷样本集中随机 选取目标缺陷图片； 将所述目标缺陷图片以预设方式进行旋转， 得到 旋转后的目标缺陷图片； 从所述正常样本图片上随机选取目标位置， 将所述旋转后的目标缺陷 图片覆盖到所述 目标位置上。 7.如权利要求6所述的数据增强方法， 其特征在于， 所述将所述目标缺陷图片以预设方 式进行旋转， 得到 旋转后的目标缺陷图片， 包括： 确定所述预设方式 中的旋转角度和旋转方向； 根据所述旋转角度和旋转方向对所述目标缺陷图片中各个像素进行位置映射， 得到旋 转后的目标缺陷图片。 8.一种数据增强装置， 其特 征在于， 包括：权　利　要　求　书 1/2 页 2 CN 114638294 A 2图片生成模块， 用于获取缺陷样本 图片， 根据所述缺陷样本 图片中的缺陷区域生成缺 陷图片， 并将所述 缺陷图片存 储于预设的缺陷样本集中； 图片覆盖模块， 用于若所述缺陷样本集中的缺陷图片满足预设条件， 则将所述缺陷样 本集中的缺陷图片随机覆盖到预设数量的正常样本图片上， 得到预设数量的目标样本图 片； 训练模块， 用于根据所述缺陷样本 图片、 所述预设数量的正常样本 图片和所述预设数 量的目标样本图片对预设的模型进行训练。 9.一种终端设备， 包括存储器、 处理器以及存储在所述存储器中并可在所述处理器上 运行的计算机程序， 其特征在于， 所述处理器执行所述计算机程序时实现如权利要求1至7 任一项所述的一种数据增强方法的步骤。 10.一种计算机可读存储介质， 所述计算机可读存储介质存储有计算机程序， 其特征在 于， 所述计算机程序被处理器执行时实现如权利要求 1至7任一项 所述的一种数据增强方法 的步骤。权　利　要　求　书 2/2 页 3 CN 114638294 A 3