[发明专利]行动对象追踪方法及相关设备

权利要求书

1.一种行动对象追踪方法，其特征在于，包括：

对当前帧图像中的第一目标检测框进行对象关键点检测以得到多个对象关键点；

根据所述多个对象关键点从所述第一目标检测框中提取出目标对象的上半部分图像，并根据所述目标对象的上半部分图像进行对象的下半部分补零处理以得到第二目标检测框，其中，所述第二目标检测框包括所述目标对象的上半部分图像以及补零后的下半部分图像；

根据所述第二目标检测框进行行动对象追踪。

2.根据权利要求1所述的方法，其特征在于，所述根据所述多个对象关键点从所述第一目标检测框中提取出目标对象的上半部分图像，并根据所述目标对象的上半部分图像进行对象的下半部分补零处理以得到第二目标检测框，包括：

将所述多个对象关键点与预设对象关键点进行比对，以确定所述第一目标检测框中的所述目标对象的下半部分图像是否完整；

若所述目标对象的下半部分图像不完整，则根据所述多个对象关键点的位置从所述第一目标检测框中提取出所述目标对象的上半部分图像；

在所述目标对象的上半部分图像的行方向上填充i行零，得到所述第二目标检测框，其中，所述i为所述第一目标检测框与所述目标对象的上半部分图像的高的差值。

3.根据权利要求1所述的方法，其特征在于，所述根据所述第二目标检测框进行行动对象追踪，包括：

将所述第二目标检测框输入预先训练好的对象特征模型，得到所述目标对象的目标对象特征；

获取j个历史对象特征，其中，所述j个历史对象特征为追踪流程中的历史对象的对象特征，所述j为正整数；

将所述目标对象特征与所述j个历史对象特征分别进行相似度计算，得到j个相似度；

若所述j个相似度中的最大相似度大于预设相似度阈值，则确定所述目标对象为历史对象。

4.根据权利要求1-3中任一项所述的方法，其特征在于，所述方法还包括：

将所述当前帧图像输入预先训练好的对象检测模型，得到所述当前帧图像对应的N个第一检测框，其中，所述N为正整数；

将对应的置信度大于预设置信度阈值的第一检测框从所述N个第一检测框中筛选出，得到M个第二检测框，其中，所述M为小于或等于所述N的正整数；

从所述M个第二检测框中获取所述第一目标检测框。

5.根据权利要求4所述的方法，其特征在于，所述从所述M个第二检测框中获取所述第一目标检测框，包括：

在所述当前帧图像中确定追踪区域；

从所述M个第二检测框中剔除不处于所述追踪区域内的第二检测框，以得到P个第三检测框，其中，所述P为小于或等于M的正整数；

从所述P个第三检测框中获取所述第一目标检测框。

6.根据权利要求4所述的方法，其特征在于，在所述将所述当前帧图像输入预先训练好的对象检测模型，得到所述当前帧图像对应的N个第一检测框之前，所述方法还包括：

判断所述当前帧图像的尺寸是否为预定尺寸；

若所述当前帧图像的尺寸不为预定尺寸，则对所述当前帧图像进行处理以使所述当前帧图像的尺寸为所述预定尺寸。

7.根据权利要求6所述的方法，其特征在于，所述对所述当前帧图像进行处理以使所述当前帧图像的尺寸为所述预定尺寸，包括：

在高维度上，若所述当前帧图像的高大于所述预定尺寸的高，则将所述当前帧图像的高缩放为所述预定尺寸的高；若所述当前帧图像的高小于所述预定尺寸的高，则在所述当前帧图像的行方向上填充h行零，其中，所述h为所述预定尺寸的高与所述当前帧图像的高的差值；

在宽维度上，若所述当前帧图像的宽大于所述预定尺寸的宽，则将所述当前帧图像的宽缩放为所述预定尺寸的宽；若所述当前帧图像的宽小于所述预定尺寸的宽，则在所述当前帧图像的列方向上填充w列零，其中，所述w为所述预定尺寸的宽与所述当前帧图像的宽的差值。