[发明专利]单目标跟踪方法及相关设备

权利要求书

1.一种单目标跟踪方法，其特征在于，应用于电子设备，所述方法包括：

在同一目标视频文件中获取第一图像和第二图像，并获取所述第一图像的第一跟踪框，其中，所述第一图像为所述第二图像的前预设帧图像，所述第一图像和所述第二图像包括同一跟踪目标，所述第一跟踪框用于框选所述第一图像中的所述跟踪目标；

将所述第一图像与所述第二图像进行叠加合成，得到目标图像；

将所述目标图像输入沙漏网络模型进行特征提取，输出目标特征图；

将所述目标特征图输入到预测网络，以输出热力图、宽高数值集和细微偏差数值集；

基于所述热力图、所述宽高数值集、所述细微偏差数值集和所述第一跟踪框，确定第二跟踪框，所述第二跟踪框用于框选所述第二图像中的所述跟踪目标；

所述预测网络包括热力图分支、宽高分支和细微偏差分支，所述将所述目标特征图输入到预测网络，以输出热力图、宽高数值集和细微偏差数值集，包括：

将所述目标特征图输入到所述热力图分支，以输出热力图，以及将所述目标特征图输入到所述宽高分支，以输出宽高数值集，以及将所述目标特征图输入到所述细微偏差分支，以输出细微偏差数值集；

其中，所述将所述目标特征图输入到所述宽高分支，以输出宽高数值集，包括：将所述目标特征图、所述第一跟踪框的第一宽度和所述第一跟踪框的第一高度输入到所述宽高分支，以输出宽高数值集。

2.根据权利要求1所述的方法，其特征在于，所述将所述第一图像与所述第二图像进行叠加合成，得到目标图像，包括：

确定所述第一图像的第一颜色通道，以及所述第二图像的第二颜色通道；

通过对所述第一颜色通道和所述第二颜色通道进行叠加，以合成目标图像。

3.根据权利要求1或2所述的方法，其特征在于，所述沙漏网络模型由i个沙漏网络依次排列构成，第i个沙漏网络的输入图像为第i-1个沙漏网络的输入图像与输出图像合成得到的图像，所述i为大于或等于2的整数；

每经过一个所述沙漏网络均进行第一处理，在所述第一处理中：将输入图像通过沙漏网络的多个第一卷积块进行下采样，输出第一特征图；将所述第一特征图通过沙漏网络的多个第二卷积块进行上采样，输出第二特征图；将所述第二特征图与所述输入图像进行叠加，输出第三特征图。

4.根据权利要求1或2所述的方法，其特征在于，所述基于所述热力图、所述宽高数值集、所述细微偏差数值集和所述第一跟踪框，确定第二跟踪框，包括：

基于所述热力图、所述细微偏差数值集和所述第一跟踪框的第一中心点的第一位置，确定第二跟踪框的第二中心点的第二位置；

基于所述宽高数值集、所述第一跟踪框的第一高度和所述第一跟踪框的第一宽度，确定所述第二跟踪框的第二高度和所述第二跟踪框的第二宽度。

5.根据权利要求4所述的方法，其特征在于，所述目标特征图包括M个特征点，M为正整数；所述基于所述热力图、所述细微偏差数值集和所述第一跟踪框的第一中心点的第一位置，确定第二跟踪框的第二中心点的第二位置，包括：

基于所述热力图，确定目标特征点在所述目标特征图中的第三位置，以及基于所述细微偏差 数值集确定所述目标特征点的偏移量，所述目标特征点为所述第一中心点与所述第二中心点连线的中点，所述M个特征点包括所述目标特征点，所述偏移量用于表示所述目标特征点相对于在所述目标特征图中的位置的偏移；

基于所述第一位置、所述第三位置和所述偏移量，确定所述第二位置。