一种样本数据的生成方法及装置制造方法及图纸

本申请提供一种样本数据的生成方法及装置，涉及物体识别技术领域，该样本数据的生成方法包括：获取样本三维模型；通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片；确定所述拍摄图片中的标记框和标记框的标记信息；在所述拍摄图片中，截取与所述标记框相对应的样本图片；对所述标记信息和所述样本图片进行关联，得到样本数据。可见，实施这种实施方式，能够通过样本数据的生成方法自动生成准确有效的样本数据集合，从而提高样本数据的可靠性，同时减少传统方法中的资源消耗与时间消耗。耗。耗。

【技术实现步骤摘要】
一种样本数据的生成方法及装置


[0001]本申请涉及物体识别领域，具体而言，涉及一种样本数据的生成方法及装置。

技术介绍

[0002]随着物品识别领域的快速发展，越来越多的识别模型被投入到生产生活中进行使用。然而，在实践中发现，现有的识别模型通常会使用小数据集来进行模型训练，从而使得最终获取到的识别模型的精确度受到极大的限制，影响识别模型的有效使用。但是，如果增加数据规模，又将会因为数据的预先标注问题，而导致工作量大大增加，从而需要消耗相当多的资源与时间。

技术实现思路

[0003]本申请实施例的目的在于提供一种样本数据的生成方法及装置，能够通过样本数据的生成方法自动生成准确有效的样本数据集合，从而提高样本数据的可靠性，同时减少传统方法中的资源消耗与时间消耗。
[0004]本申请实施例第一方面提供了一种样本数据的生成方法，包括：
[0005]获取样本三维模型；
[0006]通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片；
[0007]确定所述拍摄图片中的标记框和标记框的标记信息；
[0008]在所述拍摄图片中，截取与所述标记框相对应的样本图片；
[0009]对所述标记信息和所述样本图片进行关联，得到样本数据。
[0010]在上述实现过程中，该方法可以优先获取待拍摄的样本三维模型，然后将该样本三维模型放置在设置有虚拟相机的虚拟三维环境当中，并由虚拟相机对该样本三维模型进行图片拍摄，得到拍摄图片；在获取到拍摄图片之后，计算拍摄图片中的标记框的位置，同时对标记框中的图像进行截取得到样本图片，并且将该样本图片和标记框位置组合存储在一起，得到样本数据；其中，当样本三维模型为猫咪时，拍摄图片为某一特定方向特定角度的平面图片，如猫咪的正面全身图片，而样本图片为该平面图片中的一部分，如猫咪的脸部图片，相应的，标记信息为该脸部图片相对于全身图片的相对位置信息。可见，实施这种实施方式，能够获取到待拍摄样本的一个样本数据，基于此还能够获取到更多的不同的样本数据，从而使得该方法能够自动生成准确有效的样本数据集合，进而提高样本数据的可靠性，同时减少传统方法中的资源消耗与时间消耗。
[0011]进一步地，所述获取样本三维模型的步骤包括：
[0012]判断样本是否符合预设的体积限制要求和预设的位置摆放要求；
[0013]当所述样本符合预设的体积限制要求和预设的位置摆放要求时，获取所述样本的初始三维模型；
[0014]对所述初始三维模型进行模型毛边去除处理，得到样本三维模型。
[0015]在上述实现过程中，由于样本的体积通常不应过大，因此需要判断是否符合预设
的体积限制要求；同样的，在对样本拍摄时初始三维模型通常需要放置在合适的位置，以使初始三维模型尽可能地显露所有可见区域，因此需要判断是否符合预设的位置摆放要求；在样本满足上述要求时，根据样本生成初始三维模型，并对初始三维模型进行修正，去掉毛边，得到最终的样本三维模型。可见，实施这种实施方式，能够自动生成符合要求的样本三维模型，从而提高样本三维模型的生成精度，进而能够通过对高精度的样本三维模型进行处理得到高精度的样本数据，提高样本数据的获取精度。
[0016]进一步地，所述通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片的步骤包括：
[0017]根据预设的相机拍摄焦距、预设的图片拍摄尺寸、预设的拍摄距离、预设的拍摄角度条件以及预设的光照强度条件对虚拟相机进行多维设置；
[0018]通过完成多维设置的所述虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片。
[0019]在上述实现过程中，该方法可以预先接收相机输入参数(如焦距、图片尺寸等)、距样本三维模型的拍摄距离范围参数，同时根据对拍摄背景的要求、对拍摄角度范围的要求以及对光照变化范围的要求对虚拟相机进行参数设置与相关调整，并在完成上述设置与调整之后对样本三维模型进行拍摄，得到拍摄图片。可见，实施这种实施方式，能够对虚拟相机进行相关设置，从而能够有效地提高对拍摄效果，进而获取到有效的拍摄图片。
[0020]进一步地，所述对所述标记信息和所述样本图片进行关联，得到样本数据的步骤包括：
[0021]获取所述标记信息包括的标记框位置信息；所述标记框位置信息为所述标记框在所述拍摄图片中的位置信息；
[0022]对所述标记框位置信息和所述样本图片进行同名异格式关联，得到样本数据。
[0023]在上述实现过程中，该方法可以优先获取拍摄图片中标记框位置信息，然后将该位置信息以xml文件的形式保存到本地，同时将jpg文件形式的样本图片保存到本地；其中，样本图片和位置信息两个文件的文件名相同。可见，实施这种实施方式，能够将样本图片、标记框位置信息关联存储至指定位置，从而使得一张样本图片对应一个标记框位置信息，进而使得每一个标记框位置信息都能够表示其对应的样本图片相对于拍摄图片的位置。
[0024]进一步地，所述方法包括：
[0025]根据所述样本数据判断是否满足数据的获取结束条件；
[0026]当未满足所述获取结束条件时，执行所述的通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片的步骤。
[0027]在上述实现过程中，该方法可以设置样本获取的结束条件，从而使得样本数据获取的数量不够时，自动重新获取样本数据，进而使得该方法能够自动地获取到足够的样本数据。
[0028]本申请实施例第二方面提供了一种样本数据的生成装置，所述样本数据的生成装置包括：
[0029]获取单元，用于获取样本三维模型；
[0030]拍摄单元，用于通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片；
[0031]计算单元，用于确定所述拍摄图片中的标记框和标记框的标记信息；
[0032]标记单元，用于在所述拍摄图片中，截取与所述标记框相对应的样本图片；
[0033]存储单元，用于对所述标记信息和所述样本图片进行关联，得到样本数据。
[0034]在上述实现过程中，该样本数据的生成装置可以通过多个单元进行协同工作以获取到待拍摄样本的一个样本数据；并且，基于此还能够获取到更多的不同的样本数据，从而使得该装置能够自动生成准确有效的样本数据集合，进而提高样本数据的可靠性，同时减少传统方法中的资源消耗与时间消耗。
[0035]进一步地，所述获取单元包括：
[0036]判断子单元，用于判断样本是否符合预设的体积限制要求和预设的位置摆放要求；
[0037]获取子单元，用于当所述样本符合预设的体积限制要求和预设的位置摆放要求时，获取所述样本的初始三维模型；
[0038]修正子单元，用于对所述初始三维模型进行模型毛边去除处理，得到样本三维模型。
[0039]在上述实现过程中，该装置能够自动生成符合要求的样本三维模型，从而提高样本三维模型的生成精度，进而能够通过对高精度的样本三维模型进行处理得到高精度的样本数据，提高样本数据的获取精度本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种样本数据的生成方法，其特征在于，包括：获取样本三维模型；通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片；确定所述拍摄图片中的标记框和标记框的标记信息；在所述拍摄图片中，截取与所述标记框相对应的样本图片；对所述标记信息和所述样本图片进行关联，得到样本数据。2.根据权利要求1所述的样本数据的生成方法，其特征在于，所述获取样本三维模型的步骤包括：判断样本是否符合预设的体积限制要求和预设的位置摆放要求；当所述样本符合预设的体积限制要求和预设的位置摆放要求时，获取所述样本的初始三维模型；对所述初始三维模型进行模型毛边去除处理，得到样本三维模型。3.根据权利要求1所述的样本数据的生成方法，其特征在于，所述通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片的步骤包括：根据预设的相机拍摄焦距、预设的图片拍摄尺寸、预设的拍摄距离、预设的拍摄角度条件以及预设的光照强度条件对虚拟相机进行多维设置；通过完成多维设置的所述虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片。4.根据权利要求1所述的样本数据的生成方法，其特征在于，所述对所述标记信息和所述样本图片进行关联，得到样本数据的步骤包括：获取所述标记信息包括的标记框位置信息；所述标记框位置信息为所述标记框在所述拍摄图片中的位置信息；对所述标记框位置信息和所述样本图片进行同名异格式关联，得到样本数据。5.根据权利要求1所述的样本数据的生成方法，其特征在于，所述方法包括：根据所述样本数据判断是否满足数据的获取结束条件；当未满足所述获取结束条件时，执行所述的通过虚拟相机对所述样本三维模型进行拍摄，得到拍摄图片的步...

【专利技术属性】
技术研发人员：张二阳，刘帅，敖亚磊，侯晓龙，王俊伟，
申请(专利权)人：郑州捷安高科股份有限公司，
类型：发明
国别省市：