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

权利要求书

1.一种主体追踪方法，其特征在于，包括：

在视频流中获取一帧图像作为参考图像；

生成与所述参考图像对应的中心权重图，其中，所述中心权重图所表示的权重值从中心到边缘逐渐减小；

将所述参考图像和所述中心权重图输入至主体检测模型中，得到主体区域置信度图；

根据所述主体区域置信度图确定所述参考图像中的主体，并获取所述主体所在的主体区域；

依次获取所述视频流中所述参考图像之后的每一帧图像；

基于所述主体区域，通过追踪算法对所述参考图像之后的每一帧图像进行追踪，得到所述主体在每一帧图像中的区域；

当追踪的图像帧数大于或等于帧数阈值时，将获取的下一帧图像作为所述参考图像，返回执行对所述参考图像进行主体检测，获得所述参考图像中主体所在的主体区域的步骤。

2.根据权利要求1所述的方法，其特征在于，所述基于所述主体区域，通过追踪算法对所述参考图像之后的每一帧图像进行追踪，得到所述主体在每一帧图像中的区域，包括：

获取所述主体在上一帧图像中的区域；

将所述主体在上一帧图像中的区域增大预设尺寸，得到第一预测区域；

从当前帧图像中获取与所述第一预测区域的位置相对应的第二预测区域；

对所述第二预测区域进行追踪，得到所述主体在所述当前帧图像中的区域。

3.根据权利要求2所述的方法，其特征在于，所述从当前帧图像中获取与所述第一预测区域的位置相对应的第二预测区域，包括：

根据所述第一预测区域在所述上一帧图像中的位置将所述第一预测区域映射至所述当前帧图像，得到所述第二预测区域；和/或

获取所述第一预测区域在所述上一帧图像中的坐标位置，根据所述坐标位置从所述当前帧图像中获取对应的所述第二预测区域。

4.根据权利要求2所述的方法，其特征在于，所述将所述主体在上一帧图像中的区域增大预设尺寸，得到第一预测区域之前，还包括：

获取所述主体在所述上一帧图像之前的多帧图像中的区域；

根据所述主体在多帧所述图像中的区域分析所述主体的移动速度；

当所述移动速度大于或等于预设速度时，增大所述预设尺寸；

当所述移动速度小于所述预设速度时，减小所述预设尺寸。

5.根据权利要求1所述的方法，其特征在于，所述当追踪的图像帧数大于或等于帧数阈值时，将获取的下一帧图像作为所述参考图像之前，还包括：

获取所述主体在已追踪的多帧图像中的区域；

基于所述主体在已追踪的多帧图像中的区域分析所述主体的位置变化量，其中，所述位置变化量表示所述主体在图像中的位置变化幅度；

当所述位置变化量大于或等于变化量阈值时，将所述帧数阈值设为第一数值；

当所述位置变化量小于所述变化量阈值时，将所述帧数阈值设为第二数值，其中，所述第二数值大于所述第一数值。

6.根据权利要求1所述的方法，其特征在于，应用于电子设备，所述电子设备包含陀螺仪，所述方法还包括：

获取所述陀螺仪输出的角速度数据；

根据所述角速度数据分析所述电子设备的抖动幅度；

根据所述抖动幅度对所述帧数阈值进行调整。

7.根据权利要求1所述的方法，其特征在于，所述基于所述主体区域，通过追踪算法对所述参考图像之后的每一帧图像进行追踪，得到所述主体在每一帧图像中的区域，包括：

获取所述参考图像中每一个所述主体对应的主体区域和类别；

根据每一个所述主体对应的类别的优先等级、主体区域的大小和主体区域的位置中的至少一种确定每一个所述主体的追踪顺序；

基于所述追踪顺序对所述参考图像之后的每一帧图像进行追踪，得到每一帧图像中每一个所述主体所在的区域。