[发明专利]一种数据集采集方法、系统以及电子装置

说明书

本发明涉及数据采集技术领域，尤其涉及一种数据集采集方法、系统以及电子装置。包括如下步骤：提供一个目标；对所述目标进行拍摄获得视频；读取所述视频中每一帧图像，并获得目标在每一帧图像中的位置信息和/或特征信息；对目标进行标注，通过标准化的数据采集流程，并行化采集任务，使得数据采集和标注效率有了巨大的提升，解决了人工标注成本高、耗时长、不准确等问题，适用在实际工业应用场景中需要快速采集一批带有准确标签的图像数据的问题。

【技术领域】

本发明涉及数据采集技术领域，尤其涉及一种数据集采集方法、系统以及电子装置。

【背景技术】

如今，货物运输、货物生产以及货物销售等行业都需要基于货物本身的特征信息对不同种类的货物进行分类以及检测，因此，当面对需要检测和分类的大量货物时，通常都需要手工采集关于每个货物多方位、不同场景下的多种特征数据信息，并利用这些特征数据信息对货物进行标注，这样的操作往往效率比较低，成本比较高，并且容易出错。

【发明内容】

针对现有数据采集方法中存在的问题，本发明提供一种数据集采集方法、系统以及电子装置。

本发明提供一种数据集采集方法，其包括如下步骤：

S1、提供一个目标，所述目标为货物；S2、对所述目标进行拍摄获得视频；S3、读取所述视频中每一帧图像，并获得目标在每一帧图像中的位置信息和/或特征信息；所述步骤S3具体包括如下步骤：S31、标定视频上第一帧中目标的初始位置并获得目标的位置信息和特征信息；S32、读取下一帧视频图像，采用目标追踪算法基于该帧图像的特征信息结合上一帧中目标的位置信息和特征信息计算获得多个可能的新位置以及置信率；及S33、目标追踪算法判断多个新位置中是否存在需要追踪的目标，若是，则更新目标位置以及更新跟踪状态同时该帧目标的位置信息和特征信息对应更新并返回步骤S32中直至视频结束；若否，则重新初始化目标位置并返回步骤S32；及S4、使用代表所述目标的种类信息对所述目标进行标注，以形成与每个目标对应的数据集。

优选地，所述步骤S3中目标在每一帧中的特征信息包括颜色特征信息，所述特征信息用多维向量表示。

优选地，目标追踪算法判断多个新位置中是否存在需要追踪的目标具体步骤为：结合多个新位置确定的区域的特征信息、每个新位置的位置信息和上一帧中的目标比对以及新位置的置信率去判断目标是否存在新位置上。

优选地，目标追踪算法结合多个新位置确定的区域的特征信息、每个新位置的位置信息和上一帧中的目标比对以及新位置的置信率去判断目标是否存在新位置上包括步骤：通过分别比对新位置确定的区域的特征信息和上一帧目标的特征信息的差值是否在设定的阈值范围内，每个新位置的位置信息和上一帧目标的位置信息的差值是否在设定的阈值范围内，以及新位置的置信率是否在设定的阈值范围内。

优选地，上述步骤S2中，对所述目标进行拍摄获得视频是在预设的拍摄条件下进行，预设的拍摄条件包括设定的拍摄角度、光照条件和背景条件。

本发明为了解决上述技术问题还提供一种数据集采集系统，其包括拍摄模块和数据读取模块，所述拍摄模块用于对目标进行视频拍摄；所述数据读取模块用于读取视频中每一帧图像并获得目标在每一帧图像中的位置信息和/或特征信息；所述数据读取模块包括：获取模块、运算模块、追踪模块和第一判断模块，所述获取模块用于获取每一帧图像的特征信息以及位置信息，所述运算模块用于基于上一帧中目标的特征信息计算获得后续帧中与目标相关的多个新位置并输出每个新位置对应的置信率；所述追踪模块：用于基于运算模块获得的多个新位置对目标进行追踪；所述第一判断模块：用于判断运算模块获得的多个新位置中是否存在需要追踪的目标。

优选地，所述数据集采集系统还包括存储模块，用于对标注好的图像进行数据存储。