[发明专利]一种数据增广的方法、装置及存储介质

权利要求书

1.一种数据增广的方法，其特征在于，包括：

获取包含跟踪目标的原始样本集；

根据所述跟踪目标的运动偏好，以及外界环境对所述跟踪目标的影响，对所述原始样本集中的原始样本图像进行变换，获得与所述外界环境影响对应的第一样本集和与所述运动偏好对应的第二样本集；

将所述原始样本集和所述第一样本集，以及所述第二样本集组成的集合，作为所述跟踪目标的目标跟踪模型的训练样本集。

2.如权利要求1所述的方法，其特征在于，根据所述跟踪目标的运动偏好，以及外界环境对所述跟踪目标的影响，对所述原始样本集中的原始样本图像进行变换，获得与所述外界环境影响对应的第一样本集和与所述运动偏好对应的第二样本集，包括：

根据所述外界环境的光照发生变化的概率，从所述原始样本集中随机选取与所述光照发生变化的概率对应的第一部分原始样本集的原始样本图像，进行伽马校正，获得校正后的样本集；

根据所述外界环境中遮挡物对所述跟踪目标的遮挡概率，从所述原始样本集中随机选取与所述遮挡概率对应的第二部分原始样本集的原始样本图像，进行遮挡处理，获得遮挡后的样本集；

将所述校正后的样本集与所述遮挡后的样本集组成的集合，作为所述第一样本集；

根据所述跟踪目标的运动偏好，对所述原始样本集中的原始样本图像进行空间变换，获得所述第二样本集。

3.如权利要求2所述的方法，其特征在于，从所述原始样本集中随机选取与所述遮挡概率对应的第一部分原始样本集的原始样本图像，进行遮挡处理，获得遮挡后的样本集，包括：

确定遮挡块的大小和色彩，及在所述第一部分原始样本集的原始样本图像中的放置位置；

将具有所述大小和所述颜色的遮挡块叠加到所述第一部分原始样本集的原始样本图像中的放置位置，获得所述遮挡后的样本集。

4.如权利要求3所述的方法，其特征在于，确定遮挡块的大小和色彩，及在所述第一部分原始样本集的原始样本图像中的放置位置，获得所述遮挡后的样本集，包括：

根据所述原始样本图像的面积及所述遮挡块的面积控制因子，确定所述遮挡块的面积；其中，所述面积控制因子为所述遮挡块的面积与所述原始样本图像的面积的比率；

根据所述遮挡块的形状控制因子，以及所述遮挡块的面积，确定所述遮挡块的长和宽；所述形状控制因子为常数；

根据所述长和宽，以及所述放置位置，确定所述遮挡块在所述原始样本图像中的区域；

将所述区域中的每个像素的数据替换为对应色彩的数据，获得所述遮挡后的样本集中对应的图像。

5.如权利要求2所述的方法，其特征在于，根据所述跟踪目标的运动偏好，对所述原始样本集中的原始样本图像进行空间变换，获得所述第二样本集，包括：

将所述跟踪目标的每种运动方向出现的概率，设置为所述每种运动方向对应空间变换的占比率；其中，所述跟踪目标的每种运动方向出现的概率用于表征所述运动偏好；

针对所述每种运动方向，从所述原始样本集中随机选取与所述占比率对应的第三部分原始样本集中的原始样本图像进行对应的空间变化，获得所述每种运动方向对应的样本集；

将所有运动方向的样本集组成的集合作为所述第一样本集。

6.如权利要求1-5任一项所述的方法，其特征在于，所述空间变换，包括：

图像水平翻转、图像垂直翻转、图像随机角度旋转。

7.如权利要求6所述的方法，其特征在于，当所述空间变换为所述图像随机角度旋转时，从所述原始样本集中随机选取与所述占比率对应的部分原始样本集中的原始样本图像进行对应的空间变化，获得所述每种运动方向对应的样本集，包括：

以所述部分原始样本集中的原始样本图像的中心为旋转中心，对对应的原始样本图像进行旋转，获得旋转后的原始样本图像；

将所述旋转后的原始样本图像中超出对应原始样本图像的范围的像素数据被丢弃，并将所述旋转后的原始样本图像中无数据的像素用指定数据填补，获得与所述图像随机角度旋转对应运动方向的样本集。