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

说明书

本申请公开了一种跟踪目标确定方法及相关设备，应用于电子设备，方法包括：在同一目标视频文件中获取第一图像和第二图像，并获取第一图像的N个第一跟踪框，第一图像为第二图像的前预设帧图像，第一图像和第二图像均包括N个跟踪目标；将第二图像输入沙漏网络模型进行特征提取，输出目标特征图；将目标特征图输入到预测网络，以输出热力图、宽高数值集和特征向量集；基于热力图和宽高数值集，确定N个第二跟踪框；基于N个第一跟踪框、N个第二跟踪框和特征向量集，确定N个第一对应关系；基于N个第一对应关系确定N个第二跟踪框框选的跟踪目标。采用本申请实施例可提升确定跟踪目标的准确度。

技术领域

本申请涉及电子技术领域，尤其涉及一种跟踪目标确定方法及相关设备。

背景技术

目标跟踪是图像处理和视频处理领域的关键技术之一。目标跟踪用于识别视频或图像中的跟踪目标，广泛应用于智慧交通、人机交互及国防侦查等相关领域。跟踪目标的确定是实现目标跟踪必不可少的关键步骤之一，目前实现跟踪目标的确定主要是采用deepsort(深度排序)算法，而使用deep sort仅利用预测的位置信息进行匹配，预测准确度低。

发明内容

本申请实施例提供一种跟踪目标确定方法及相关设备，用于提升确定跟踪目标的准确度。

第一方面，本申请实施例提供一种跟踪目标确定方法，应用于电子设备，所述方法包括：

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

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

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

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

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

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

第二方面，本申请实施例提供一种跟踪目标确定装置，应用于电子设备，该装置包括：

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

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

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

跟踪框确定单元，用于基于所述热力图和所述宽高数值集，确定N个第二跟踪框；

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

跟踪目标确定单元，用于基于所述N个第一对应关系确定所述N个第二跟踪框框选的所述跟踪目标。

第三方面，本申请实施例提供一种电子设备，包括处理器、存储器、通信接口以及一个或多个程序，其中，上述一个或多个程序被存储在上述存储器中，并且被配置由上述处理器执行，上述程序包括用于执行本申请实施例第一方面所述的方法中的步骤的指令。