[发明专利]多目标检测与跟踪方法、装置、计算机设备和存储介质

说明书

本申请涉及一种多目标检测与跟踪方法、装置、计算机设备、存储介质和计算机程序产品。所述方法包括：获取卫星遥感影像的目标影像图集，将目标影像图集中预设帧数的连续影像图输入预先训练好的特征提取模型中，对连续影像图进行多检测目标特征提取操作，得到各检测目标的目标预测头与背景特征；查找连续影像图中目标图像帧的各跟踪目标的跟踪特征；基于各检测目标的目标检测头与背景特征，以及各跟踪目标的跟踪特征，确定各检测目标与各跟踪目标的关联性；根据各检测目标与各跟踪目标的关联性得到各检测目标的位置信息与跟踪轨迹。采用本方法能够有效提高了多目标检测的准确性及速度，进而提升了多目标的检测跟踪性能。

技术领域

本申请涉及数据处理技术领域，特别是涉及一种多目标检测与跟踪方法、装置、计算机设备、存储介质和计算机程序产品。

背景技术

近年来，商业航天与卫星遥感产业发展迅猛，卫星遥感技术目前已广泛应用于港口船舶流量分析、交通流量分析、机场航班流量分析等多个应用领域，而基于卫星遥感数据对多目标进行检测和跟踪是这些应用领域的重要任务。

不同于传统多目标检测与跟踪场景，在卫星遥感视频影像中，其检测追踪主要难点在于目标尺度小、背景噪声大，因此传统检测器的准确率较低，造成大量漏检与误检，影响跟踪性能。

发明内容

基于此，有必要针对上述技术问题，提供一种能够有效提升检测准确性和跟踪性能的多目标检测与跟踪方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。

第一方面，本申请提供了一种多目标检测与跟踪方法，所述方法包括：

获取卫星遥感影像的目标影像图集，将所述目标影像图集中预设帧数的连续影像图输入预先训练好的特征提取模型中，对所述连续影像图进行多检测目标特征提取操作，得到各所述检测目标的目标预测头与背景特征；

查找所述连续影像图中目标图像帧的各跟踪目标的跟踪特征；

基于各所述检测目标的目标检测头与背景特征，以及各所述跟踪目标的跟踪特征，确定各所述检测目标与各所述跟踪目标的关联性；

根据各所述检测目标与各跟踪目标的关联性得到各所述检测目标的位置信息与跟踪轨迹。

在其中一个实施例中，所述获取卫星遥感影像的目标影像图集，包括：

获取卫星遥感影像，对所述卫星遥感影像进行分帧处理，得到包含了连续帧遥感影像图的影像图集；

将所述影像图集中各帧影像图按照预设切分规则进行切分，得到切分后的各目标影像图；

根据各所述目标影像图对应显示的遥感区域进行分类，得到所述卫星遥感影像的目标影像图集，所述目标影像图集与所述遥感区域对应。

在其中一个实施例中，所述对所述连续影像图进行多检测目标特征提取操作，得到各所述检测目标的目标预测头与背景特征，包括：

通过所述特征提取模型中的第一网络对所述连续影像图的目标图像帧进行特征提取操作，得到所述目标图像帧的前后背景特征；

通过所述特征提取模型中的第二网络对所述连续影像图进行特征提取操作，得到所述连续影像图的时序特征；

将所述目标图像帧的前后背景特征与所述连续影像图的时序特征进行融合，得各所述检测目标的目标预测头；

将所述连续影像图的时序特征进行分层融合，得到各所述检测目标的背景特征。

在其中一个实施例中，所述基于各所述检测目标的目标检测头与背景特征，以及各所述跟踪目标的跟踪特征，确定各所述检测目标与各所述跟踪目标的关联性，包括：

基于各所述检测目标的目标预测头，得到各所述检测目标的检测坐标和目标外观特征；