[发明专利]目标检测数据集生成方法、装置、系统和电子设备

说明书

本发明提供了一种目标检测数据集生成方法、装置、系统和电子设备，通过获取所述采集台采集的多张不同角度的工件的第一图片；提取每张所述第一图片中的工件图像；把所述工件图像放入预存的背景图片中，得到第二图片；根据工件图像的位置和大小在所述第二图片中生成边框信息和标签；从而可快速生成目标检测数据集，工作效率高。

技术领域

本发明涉及人工智能技术领域，尤其涉及一种目标检测数据集生成方法、装置、系统和电子设备。

背景技术

在人工智能领域，基于深度学习的目标检测算法是近年来比较热门的研究课题。在把目标检测算法应用到实际工业时，因为产品迭代周期短和数据集的短缺，需要为经常更新的目标检测物体建立相对应的数据集。一般的建立目标检测的数据集的过程主要有三步：

第一步，将待检测物体放在不同的背景环境下，从尽可能多的不同角度拍照；

第二步，人工地用标签软件将待检测目标物从放置台背景用四方形框出并标签；

第三步，将被编辑的图片保存为二进制文件，形成数据集，方便以后检测算法读入。

通常情况下，为了保证训练出来模型的鲁棒性，同一个被检测的物体要有数千张到数万张不等的数据集。用人工的方法来逐张地采集照片并对数据分割和标签，会耗费大量的时间和精力，且效率太低不能满足工业产品特别是电子产品更新换代的要求。

发明内容

鉴于上述现有技术的不足之处，本申请实施例的目的在于提供一种目标检测数据集生成方法、装置、系统和电子设备，其生成目标检测数据集的效率高。

第一方面，本申请实施例提供一种目标检测数据集生成方法，应用于数据采集装置,所述数据采集装置包括采集台，所述采集台包括可旋转的转台和设置在所述转台上方且角度可调的摄像机，所述目标检测数据集生成方法包括步骤：

获取所述采集台采集的多张不同角度的工件的第一图片；

提取每张所述第一图片中的工件图像；

把所述工件图像放入预存的背景图片中，得到第二图片；

根据工件图像的位置和大小在所述第二图片中生成边框信息和标签。

所述的目标检测数据集生成方法中，所述通过所述采集台获取多张不同角度的工件的第一图片的步骤包括：

向所述采集台发送控制指令，以控制所述采集台的转台转动多个不同的角度，并在转台的每个角度下，控制所述摄像机从多个不同角度拍摄工件的第一图片。

所述的目标检测数据集生成方法中，所述提取每张所述第一图片中的工件图像的步骤之后，还包括：

对每个所述工件图像进行预处理，从而由每个所述工件图像生成多个角度和/或大小不同的工件图像。

进一步的，所述对每个所述工件图像进行预处理，从而由每个所述工件图像生成多个角度和/或大小不同的工件图像的步骤包括：

对所述工件图像进行旋转处理：把所述工件图像转动多个不同角度，每转动一个角度则保存为一个新的工件图像；

和/或

对所述工件图像进行缩放处理：把所述工件图像按多个缩放比例值进行缩放，每缩放一次则保存为一个新的工件图像。

所述的目标检测数据集生成方法中，所述把所述工件图像放入预存的背景图片中，得到第二图片的步骤包括：

把同一个工件图像在不同位置放入所述背景图片，从而每个工件图像对应得到多个第二图片。

第二方面，本申请实施例提供一种目标检测数据集生成装置，包括：