[发明专利]主体跟踪方法和装置、电子设备、计算机可读存储介质

权利要求书

1.一种主体跟踪方法，其特征在于，应用于包含至少两个摄像头的电子设备，包括：

分别控制所述至少两个摄像头实时获取图像；

从获取的图像中确定主体检测帧，并将所述主体检测帧之后获取的图像作为主体跟踪帧；

根据所述主体检测帧生成对应的中心权重图，其中，所述中心权重图所表示的权重值从中心到边缘逐渐减小；

将所述主体检测帧和对应的所述中心权重图输入到主体检测模型中，得到所述主体检测帧的主体区域置信度图，其中，所述主体检测模型是预先根据同一场景的主体检测帧、中心权重图及对应的已标注的主体掩膜图进行训练得到的模型；

根据所述主体检测帧的主体区域置信度图确定所述主体检测帧中的目标主体；

基于所述主体检测帧的目标主体，逐帧对所述主体跟踪帧进行主体跟踪。

2.根据权利要求1所述的方法，其特征在于，所述根据所述主体检测帧的主体区域置信度图确定所述主体检测帧中的目标主体，包括：

对所述主体检测帧的主体区域置信度图进行处理，得到对应的主体掩膜图；

检测所述主体检测帧，确定所述主体检测帧中的高光区域；

根据所述主体检测帧中的高光区域与对应的所述主体掩膜图，确定所述主体检测帧中消除高光的目标主体。

3.根据权利要求2所述的方法，其特征在于，所述对所述主体检测帧的主体区域置信度图进行处理，得到对应的主体掩膜图，包括：

对所述主体检测帧的主体区域置信度图进行自适应置信度阈值过滤处理，得到对应的主体掩膜图。

4.根据权利要求1所述的方法，其特征在于，所述至少两个摄像头的帧率均相同。

5.根据权利要求1所述的方法，其特征在于，所述基于所述主体检测帧的目标主体，逐帧对所述主体跟踪帧进行主体跟踪，包括：

基于所述主体检测帧的目标主体，对实时获取的当前主体跟踪帧进行主体跟踪，得到所述当前主体跟踪帧的目标主体；

基于所述当前主体跟踪帧的目标主体，对实时获取的下一帧主体跟踪帧进行主体跟踪，得到所述下一帧主体跟踪帧的目标主体；

将所述下一帧主体跟踪帧作为当前主体跟踪帧，返回执行所述基于所述当前主体跟踪帧的目标主体，对实时获取的下一帧主体跟踪帧进行主体跟踪，得到所述下一帧主体跟踪帧的目标主体步骤。

6.根据权利要求5所述的方法，其特征在于，所述基于所述主体检测帧的目标主体，对实时获取的当前主体跟踪帧进行主体跟踪，得到所述当前主体跟踪帧的目标主体，包括：

获取所述主体检测帧的目标主体的位置区域；

基于所述主体检测帧的目标主体的位置区域，获取当前主体跟踪帧的搜索区域；其中，所述搜索区域覆盖所述主体检测帧的目标主体的位置区域；

从所述当前主体跟踪帧的搜索区域中进行查找，得到所述当前主体跟踪帧的目标主体。

7.根据权利要求1所述的方法，其特征在于，所述方法还包括：

统计连续进行主体跟踪的主体跟踪帧的帧数量；

当所述帧数量达到数量阈值时，返回执行所述从获取的图像中确定主体检测帧，并将所述主体检测帧之后获取的图像作为主体跟踪帧步骤。

8.一种主体跟踪装置，其特征在于，应用于包含至少两个摄像头的电子设备，包括：

图像获取模块，用于分别控制所述至少两个摄像头实时获取图像；

确定模块，用于从获取的图像中确定主体检测帧，并将所述主体检测帧之后获取的图像作为主体跟踪帧；

主体检测模块，用于根据所述主体检测帧生成对应的中心权重图，其中，所述中心权重图所表示的权重值从中心到边缘逐渐减小；将所述主体检测帧和对应的所述中心权重图输入到主体检测模型中，得到所述主体检测帧的主体区域置信度图，其中，所述主体检测模型是预先根据同一场景的主体检测帧、中心权重图及对应的已标注的主体掩膜图进行训练得到的模型；根据所述主体检测帧的主体区域置信度图确定所述主体检测帧中的目标主体；

主体跟踪模块，用于基于所述主体检测帧的目标主体，逐帧对所述主体跟踪帧进行主体跟踪。