[发明专利]训练目标跟踪模型和目标跟踪的方法和装置

权利要求书

1.一种训练目标跟踪模型的方法，包括：

获取样本集，其中，所述样本集中的样本包括视频帧和用于标注所述视频帧中目标对象的真实框；

构建目标跟踪模型，其中，所述目标跟踪模型的头部包括交并比头，用于计算交并比损失值；

执行如下训练步骤：从所述样本集中选取样本；将选取的样本中的视频帧输入所述目标跟踪模型，输出预测框；根据选取的样本中的真实框和所述预测框的差异计算原始损失值和交并比损失值；若所述原始损失值和所述交并比损失值的加权和小于预定阈值，则确定出所述目标跟踪模型训练完成；

否则，调整所述目标跟踪模型的网络参数，继续执行所述训练步骤；

其中，所述构建目标跟踪模型，包括：

获取原始目标跟踪模型；

获取应用所述目标跟踪模型的终端的运算能力；

若所述运算能力大于第一预定能力，则使用HarDNet-85替换掉所述原始目标跟踪模型中的骨干网络，得到构建出的目标跟踪模型；

使用深度可分离卷积替换所述目标跟踪模型中的头部中的普通卷积。

2.根据权利要求1所述的方法，其中，所述构建目标跟踪模型，包括：

获取原始目标跟踪模型；

获取应用所述目标跟踪模型的终端的运算能力；

若所述运算能力小于第二预定能力，则使用HRNetV2-W18替换掉所述原始目标跟踪模型中的骨干网络，并使用深层融合特征金字塔结构替换掉所述原始目标跟踪模型中的颈部，得到构建出的目标跟踪模型。

3.根据权利要求2所述的方法，其中，所述构建目标跟踪模型，包括：

去除所述目标跟踪模型中的可变形卷积。

4.根据权利要求1所述的方法，其中，所述调整所述目标跟踪模型的网络参数，包括：

通过同步批量归一化方式和滑动平均方式调整所述目标跟踪模型的网络参数。

5.一种目标跟踪方法，包括：

获取待检测的视频帧集合；

将所述视频帧集合输入根据权利要求1-4中任一项所述的方法训练完成的目标跟踪模型，在每个视频帧中输出至少一个检测框；

对于每个视频帧，根据检测框的得分将该视频帧中的检测框划分成高分框集合和低分框集合；

对于每个视频帧，将该视频帧的高分框集合与之前已确定的跟踪轨迹进行第一次匹配，将第一次匹配失败的跟踪轨迹与该视频帧的低分框集合进行第二次匹配，得到更新后的跟踪轨迹。

6.根据权利要求5所述的方法，其中，所述第一次匹配和所述第二次匹配包括交并比匹配，所述目标跟踪模型输出检测框的阈值小于原始目标跟踪模型输出检测框的阈值。

7.一种训练目标跟踪模型的装置，包括：

获取单元，被配置成获取样本集，其中，所述样本集中的样本包括视频帧和用于标注所述视频帧中目标对象的真实框；

构建单元，被配置成构建目标跟踪模型，其中，所述目标跟踪模型的头部包括交并比头，用于计算交并比损失值；

训练单元，被配置成执行如下训练步骤：从所述样本集中选取样本；将选取的样本中的视频帧输入所述目标跟踪模型，输出预测框；根据选取的样本中的真实框和所述预测框的差异计算原始损失值和交并比损失值；若所述原始损失值和所述交并比损失值的加权和小于预定阈值，则确定出所述目标跟踪模型训练完成；

调整单元，被配置成否则，调整所述目标跟踪模型的网络参数，继续执行所述训练步骤；

其中，所述构建单元进一步被配置成：

获取原始目标跟踪模型；

获取应用所述目标跟踪模型的终端的运算能力；

若所述运算能力大于第一预定能力，则使用HarDNet-85替换掉所述原始目标跟踪模型中的骨干网络，得到构建出的目标跟踪模型；

使用深度可分离卷积替换所述目标跟踪模型中的头部中的普通卷积。