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

权利要求书

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

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

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

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

基于所述热力图和所述宽高数值集，确定N个第二跟踪框，所述N个第二跟踪框用于框选所述第二图像中N个跟踪目标；

基于所述N个第一跟踪框、所述N个第二跟踪框和所述特征向量集，确定N个第一对应关系，所述N个第一对应关系用于表征所述N个第一跟踪框与所述N个第二跟踪框的一一对应关系；

基于所述N个第一对应关系确定所述N个第二跟踪框框选的所述跟踪目标；

其中，所述基于所述热力图和所述宽高数值集，确定N个第二跟踪框，包括：基于所述热力图确定所述N个第二跟踪框的第一中心点的第一位置；基于所述宽高数值集确定所述N个第二跟踪框的第一高度和所述N个第二跟踪框的第一宽度。

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

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

3.根据权利要求1或2所述的方法，其特征在于，所述预测网络包括热力图分支、宽高分支和特征向量分支；所述将所述目标特征图输入到预测网络，以输出热力图、宽高数值集和特征向量集，包括：

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

将所述热力图输入所述特征向量分支，以输出特征向量集。

4.根据权利要求1所述的方法，其特征在于，所述基于所述N个第一跟踪框、所述N个第二跟踪框和所述特征向量集，确定N个第一对应关系，包括：

根据通过所述特征向量集以及所述N个第一中心点的第一位置，确定所述N个第一中心点的特征向量；

根据所述N个第一中心点的特征向量和所述N个第一跟踪框的第二中心点的特征向量，确定N个偏移量集，以及根据所述N个第一跟踪框和所述N个第二跟踪框，确定N个匹配度集，所述N个偏移量集与所述N个第一中心点的特征向量一一对应，每个所述偏移量集均包括N个偏移量，所述N个偏移量为其对应的所述第一中心点的特征向量，相对于任一所述第二中心点的特征向量在所述特征向量集的位置偏移，所述N个匹配度集与所述N个第一跟踪框一一对应，每个所述匹配度集均包括N个匹配度，所述N个匹配度为其对应的所述第一跟踪框与任一所述第二跟踪框的匹配度；

根据所述N个偏移量集和所述N个匹配度集确定N个第二对应关系，所述N个第二对应关系用于表征所述N个第二中心点与所述N个第一中心点的一一对应关系；

根据所述N个第二对应关系确定N个第一对应关系。