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

权利要求书

1.一种数据集采集方法，其特征在于：包括如下步骤：

S1、提供一个目标，所述目标为货物；

S2、对所述目标进行拍摄获得视频；

S3、读取所述视频中每一帧图像，并获得目标在每一帧图像中的位置信息和/或特征信息；

所述步骤S3具体包括如下步骤：

S31、标定视频上第一帧中目标的初始位置并获得目标的位置信息和特征信息；

S32、读取下一帧视频图像，采用目标追踪算法基于该帧图像的特征信息结合上一帧中目标的位置信息和特征信息计算获得多个可能的新位置以及置信率；及

S33、目标追踪算法判断多个新位置中是否存在需要追踪的目标，若是，则更新目标位置以及更新跟踪状态同时该帧目标的位置信息和特征信息对应更新并返回步骤S32中直至视频结束；若否，则重新初始化目标位置并返回步骤S32；

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

S4、使用代表所述目标的种类信息对所述目标进行标注，以形成与每个目标对应的数据集。

2.如权利要求1所述的数据集采集方法，其特征在于：所述步骤S3中目标在每一帧中的特征信息包括颜色特征信息，所述特征信息用多维向量进行表示。

3.如权利要求2所述的数据集采集方法，其特征在于：目标追踪算法结合多个新位置确定的区域的特征信息、每个新位置的位置信息和上一帧中的目标比对以及新位置的置信率去判断目标是否存在新位置上包括步骤：通过分别比对新位置确定的区域的特征信息和上一帧目标的特征信息的差值是否在设定的阈值范围内，每个新位置的位置信息和上一帧目标的位置信息的差值是否在设定的阈值范围内，以及新位置的置信率是否在设定的阈值范围内。

4.如权利要求1所述的数据集采集方法，其特征在于：上述步骤S2中，对所述目标进行拍摄获得视频是在预设的拍摄条件下进行，预设的拍摄条件包括设定的拍摄角度、光照条件和背景条件。

5.一种用于执行如权利要求1所述的数据集采集方法的数据集采集系统，其特征在于：其包括拍摄模块和数据读取模块，所述拍摄模块用于对目标进行视频拍摄；所述数据读取模块用于读取视频中每一帧图像并获得目标在每一帧图像中的位置信息和/或特征信息；所述数据读取模块包括：获取模块、运算模块、追踪模块和第一判断模块，所述获取模块用于获取每一帧图像的特征信息以及位置信息，所述运算模块用于基于上一帧中目标的特征信息计算获得后续帧中与目标相关的多个新位置并输出每个新位置对应的置信率；所述追踪模块：用于基于运算模块获得的多个新位置对目标进行追踪；所述第一判断模块：用于判断运算模块获得的多个新位置中是否存在需要追踪的目标。

6.如权利要求5所述的数据集采集系统，其特征在于：所述数据集采集系统还包括存储模块，用于对标注好的图像进行数据存储。

7.一种电子装置，包括存储器和处理器，其特征在于：所述存储器中存储有计算机程序，所述处理器被设置为通过所述计算机程序执行所述权利要求1至4任一项中所述的数据集采集方法。