[发明专利]数据集处理方法、装置、存储介质及电子设备

权利要求书

1.一种数据集处理方法，其特征在于，所述方法包括：

获取待处理数据集，所述待处理数据集包括第一类样本数据，其中，所述第一类样本为包括单个动作的样本数据；

从所述待处理数据集中随机选取至少三个不同动作类别的第一类样本数据，并将所述至少三个不同动作类别的第一类样本数据拼接为拼接样本数据；

对所述拼接样本数据进行采样，以获取采样样本数据集；

根据所述采样样本数据集对所述待处理数据集进行扩充，得到目标数据集；所述目标数据集包括所述采样样本数据集和所述待处理数据集中的所有样本。

2.根据权利要求1所述的方法，其特征在于，获取待处理数据集，包括：

获取原始数据集，并将所述原始数据集中的第二类样本数据分割为多个第一类样本，得到分割后的原始数据集；其中，所述第二类样本数据为包括多个动作的样本数据；

根据每个所述第一类样本包括的动作的动作类别对所述第一类样本进行编码，得到每个所述第一类样本对应的标签；

根据所述第一类样本及其对应的标签得到所述待处理数据集。

3.根据权利要求2所述的方法，其特征在于，获取原始数据集，并将所述原始数据集中的第二类样本数据分割为多个第一类样本，得到分割后的原始数据集，包括：

确定所述第二类样本数据中的每个动作的起始帧及结束帧；

根据所述起始帧及所述结束帧将所述第二类样本数据分割为多个第一类样本。

4.根据权利要求1所述的方法，其特征在于，对所述拼接样本数据进行采样，以获取采样样本数据集，包括：

对所述拼接样本数据进行采样，得到多个采样样本；

对每个采样样本进行编码，以获取所述采样样本的标签；

根据所述多个采样样本及编码获取采样样本数据集。

5.根据权利要求4所述的方法，其特征在于，对所述拼接样本数据进行采样，包括：

获取采样长度；

确定所述拼接样本数据的起始采样点及结束采样点；

根据所述采样长度、所述起始采样点及所述结束采样点对所述拼接样本数据进行采样。

6.根据权利要求5所述的方法，其特征在于，获取采样长度，包括：

根据所述拼接样本数据中的三个不同动作类别的第一类样本数据确定头部片段、中间片段以及尾部片段，所述头部片段、中间片段以及尾部片段分别对应不同动作类别；

根据所述中间片段的长度获取所述采样长度。

7.根据权利要求6所述的方法，其特征在于，对每个采样样本进行编码，以获取所述采样样本的标签，包括：

确定所述采样样本的中心位置与所述中间片段的中心位置之间的相对偏移；

根据所述相对偏移计算所述采样样本属于所述中间片段包括的行为类型的第一置信度；

根据所述第一置信度确定所述采样样本属于其他行为类型的第二置信度；

根据所述第一置信度及所述第二置信度对所述采样样本进行编码，得到与所述采样样本对应的标签。

8.一种数据集处理装置，其特征在于，所述装置包括：

待处理数据集获取模块，用于获取待处理数据集，所述待处理数据集包括第一类样本数据，其中，所述第一类样本为包括单个动作的样本数据；

样本拼接模块，用于从所述待处理数据集中随机选取至少三个不同动作类别的第一类样本数据，并将所述至少三个不同动作类别的第一类样本数据拼接为拼接样本数据；

采样模块，用于对所述拼接样本数据进行采样，以获取采样样本数据集；

数据扩充模块，用于根据所述采样样本数据集对所述待处理数据集进行扩充，得到目标数据集；所述目标数据集包括所述采样样本数据集和所述待处理数据集中的所有样本。

9.一种存储介质，其特征在于，该存储介质存储的计算机程序，在被一个或多个处理器执行时，用来实现如权利要求1-7中任意一项所述的数据集处理方法。

10.一种电子设备，其特征在于，包括存储器和处理器，所述存储器上存储有计算机程序，该计算机程序被所述处理器执行时，执行如权利要求1-7任意一项所述的数据集处理方法。