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

权利要求书

1.一种目标检测数据集生成方法，其特征在于，应用于数据采集装置,所述数据采集装置包括采集台，所述采集台包括可旋转的转台和设置在所述转台上方且角度可调的摄像机，所述目标检测数据集生成方法包括步骤：

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

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

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

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

2.根据权利要求1所述的目标检测数据集生成方法，其特征在于，所述通过所述采集台获取多张不同角度的工件的第一图片的步骤包括：

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

3.根据权利要求1所述的目标检测数据集生成方法，其特征在于，所述提取每张所述第一图片中的工件图像的步骤之后，还包括：

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

4.根据权利要求3所述的目标检测数据集生成方法，其特征在于，所述对每个所述工件图像进行预处理，从而由每个所述工件图像生成多个角度和/或大小不同的工件图像的步骤包括：

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

和/或

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

5.根据权利要求1所述的目标检测数据集生成方法，其特征在于，所述把所述工件图像放入预存的背景图片中，得到第二图片的步骤包括：

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

6.一种目标检测数据集生成装置，其特征在于，包括：

获取模块，用于获取采集台采集的多张不同角度的工件的第一图片；

提取模块，用于提取每张所述第一图片中的工件图像；

合成模块，用于把所述工件图像放入预存的背景图片中，得到第二图片；

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

7.根据权利要求6所述的目标检测数据集生成装置，其特征在于，还包括：

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

8.根据权利要求6所述的目标检测数据集生成装置，其特征在于，所述合成模块在把所述工件图像放入预存的背景图片中以得到第二图片时，

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

9.一种电子设备，其特征在于，包括处理器和存储器，所述存储器中存储有计算机程序，所述处理器通过调用所述存储器中存储的所述计算机程序，用于执行权利要求1-5任一项所述的目标检测数据集生成方法的步骤。

10.一种目标检测数据集生成系统，其特征在于，包括采集台和处理装置，所述采集台与所述处理装置电性连接；

所述采集台包括可旋转的转台和设置在所述转台上方且角度可调的摄像机；所述摄像机用于拍摄放置在所述转台上的工件的第一图片；

所述处理装置用于控制所述采集台采集多张不同角度的工件的第一图片，并获取多张所述第一图片，提取每张所述第一图片中的工件图像，把所述工件图像放入预存的背景图片中以得到第二图片，根据工件图像的位置和大小在所述第二图片中生成边框信息和标签。