[发明专利]卫星视频目标跟踪方法、网络的训练方法及系统

说明书

本公开提供了一种卫星视频目标跟踪方法、网络的训练方法及系统，包括：获取相邻的多个待测卫星视频帧；将多个待测卫星视频帧输入至卫星视频目标跟踪网络的视频域图映射模块中，生成全连通图，其中，全连通图包括多个节点和边，节点用于表征待测卫星视频帧中的目标对象，边用于表征两个目标对象彼此之间的关联关系；将全连通图输入至卫星视频目标跟踪网络的时空图推理模块中，得到多个目标对象各自的检测框和身份标识信息。

技术领域

本公开涉及视频处理技术领域，更具体地，涉及一种卫星视频目标跟踪方法、网络的训练方法及系统。

背景技术

随着航天技术的发展，星载遥感已经成为一种卫星视频目标跟踪方法、网络的训练方法及系统，尤其是卫星视频目标跟踪可有效辅助分析地面运动目标的态势信息，进一步提高对地观测效率。

传统的卫星视频目标跟踪方法仅适用于卫星视频数据中的单个目标，无法做到多个目标进行并行跟踪。卫星所拍摄的高分辨率视频通常覆盖大尺度场景，包含多个目标。使用传统单目标跟踪方法需要进行多次调用才能实现多目标跟踪，系统处理效率低下，且准确率不高，难以满足星载遥感系统对目标对象跟踪的高效率、高精确度以及时效性的需求。

发明内容

有鉴于此，本公开实施例提供了一种卫星视频目标跟踪方法、网络的训练方法及系统。

本公开实施例的一个方面提供了一种卫星视频目标跟踪方法，包括：

获取相邻的多个待测卫星视频帧；

将上述多个待测卫星视频帧输入至卫星视频目标跟踪网络的视频域图映射模块中，生成全连通图，其中，上述全连通图包括多个节点和边，上述节点用于表征上述待测卫星视频帧中的目标对象，上述边用于表征两个上述目标对象彼此之间的关联关系；

将上述全连通图输入至上述卫星视频目标跟踪网络的时空图推理模块中，得到多个上述目标对象各自的检测框和身份标识信息。

本公开实施例的另一个方面提供了一种卫星视频目标跟踪网络的训练方法，包括：

获取相邻的多个样本卫星视频帧和上述多个样本卫星视频帧各自的标签；

将上述多个样本卫星视频帧输入至卫星视频目标跟踪网络的视频域图映射模块中，生成样本全连通图，其中，上述样本全连通图包括多个节点和边，上述节点用于表征上述样本卫星视频帧中的目标对象，上述边用于表征两个上述目标对象彼此之间的关联关系；

将上述样本全连通图输入至上述卫星视频目标跟踪网络的时空图推理模块中，生成多个上述目标对象各自的检测框和身份标识信息；

利用多个上述目标对象各自的检测框和身份标识信息和上述多个样本卫星视频帧各自的标签训练上述卫星视频目标跟踪网络，得到经训练的卫星视频目标跟踪网络。

本公开实施例的另一个方面提供了一种卫星视频目标跟踪系统，包括：

数据获取模块，用于获取相邻的多个待测卫星视频帧；

图构建模块，用于将上述多个待测卫星视频帧输入至卫星视频目标跟踪网络的视频域图映射模块中，生成全连通图，其中，上述全连通图包括多个节点和边，上述节点用于表征上述待测卫星视频帧中的目标对象，上述边用于表征两个上述目标对象彼此之间的关联关系；

跟踪模块，用于将上述全连通图输入至上述卫星视频目标跟踪网络的时空图推理模块中，得到多个上述目标对象各自的检测框和身份标识信息。

本公开实施例的另一个方面提供了一种卫星视频目标跟踪网络的训练系统，包括：

样本获取模块，用于获取相邻的多个样本卫星视频帧和上述多个样本卫星视频帧各自的标签；