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

说明书

本申请涉及一种主体追踪方法、装置、电子设备和计算机可读存储介质。上述方法包括：在视频流中获取一帧图像作为参考图像；对参考图像进行主体检测，获得参考图像中主体所在的主体区域；依次获取视频流中参考图像之后的每一帧图像；基于主体区域，通过追踪算法对参考图像之后的每一帧图像进行追踪，得到主体在每一帧图像中的区域；当追踪的图像帧数大于或等于帧数阈值时，将获取的下一帧图像作为参考图像，返回执行对参考图像进行主体检测，获得参考图像中主体所在的主体区域的步骤。上述方法可以在主体追踪过程更新图像的主体区域，可以提高主体追踪的准确性。

技术领域

本申请涉影像技术领域，特别是涉及一种主体追踪方法、装置、电子设备和计算机可读存储介质。

背景技术

随着影像技术的发展，主体追踪技术的应用越来越广泛。目前，主体追踪技术通常依靠用户手动选取图像中的主体，进而根据该主体对后续图像进行主体追踪。然而，由于在视频流的拍摄过程中，视频流中的主体、主体的大小、主体的位置等都可能发生变化，传统的主体追踪方法往往无法准确地追踪到主体，存在主体追踪的准确性较低的问题。

发明内容

本申请实施例提供一种主体追踪方法、装置、电子设备和计算机可读存储介质，可以提高主体追踪的准确性。

一种主体追踪方法，包括：

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

对所述参考图像进行主体检测，获得所述参考图像中主体所在的主体区域；

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

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

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

一种主体追踪装置，包括：

第一获取模块，用于在视频流中获取一帧图像作为参考图像；

主体检测模块，用于对所述参考图像进行主体检测，获得所述参考图像中主体所在的主体区域；

第二获取模块，用于依次获取所述视频流中所述参考图像之后的每一帧图像；

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

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

一种电子设备，包括存储器及处理器，所述存储器中储存有计算机程序，所述计算机程序被所述处理器执行时，使得所述处理器执行如下步骤：

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

对所述参考图像进行主体检测，获得所述参考图像中主体所在的主体区域；

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

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

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

一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现如下步骤：

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