[发明专利]模板图像的更新、目标跟踪方法及装置、电子设备及介质

说明书

本申请实施例公开了一种模板图像的更新方法及装置、目标跟踪方法及装置、电子设备及存储介质。所述模板图像的更新方法，包括：根据第t+1帧图像确定所述第t+1帧图像的特征图，其中，t为正整数；根据所述第t+1帧图像的特征图，更新第t个模板图像得到第t+1个模板图像，以根据第t+1个模板图像对第t+2帧图像进行目标跟踪；其中，所述第t个模板图像，用于对所述第t+1帧图像进行目标跟踪；第t个模板图像和第t+1个模板图像中均包含要跟踪的目标。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种模板图像的更新方法及装置、目标跟踪方法及装置、电子设备及存储介质。

背景技术

目标追踪的过程中，会采集视频，然后对视频的视频帧进行逐帧处理，追踪视频中的目标。

现有技术中提供多种进行目标追踪的方法，但是在一些情况下依然会跟丢目标，导致跟踪成功率低的问题。

发明内容

有鉴于此，本发明实施例期望提供一种模板图像的更新方法及装置、目标跟踪方法及装置、电子设备及存储介质。

本发明的技术方案是这样实现的：

本申请实施例第一方面提供一种模板图像的更新方法，包括：

根据第t+1帧图像确定所述第t+1帧图像的特征图，其中，t为正整数；

根据所述第t+1帧图像的特征图，更新第t个模板图像得到第t+1个模板图像，以根据第t+1个模板图像对第t+2帧图像进行目标跟踪；其中，所述第t个模板图像，用于对所述第t+1帧图像进行目标跟踪；第t个模板图像和第t+1个模板图像中均包含要跟踪的目标。

基于上述方案，所述方法还包括：

根据所述第t+1帧图像的特征图及所述第t个模板图像的特征图，确定第t+1帧图像的各像素的分类标签及所述分类标签的分类置信度；

所述根据所述第t+1帧图像的特征图，更新第t个模板图像得到第t+1个模板图像，包括：

根据所述第t+1帧图像的特征图及所述分类置信度，更新所述第t个模板图像得到所述第t+1个模板图像。

基于上述方案，所述根据所述第t+1帧图像的特征图及所述分类置信度，更新所述第t个模板图像得到所述第t+1个模板图像，包括：

根据所述分类置信度，确定更新步长；

基于所述更新步长，确定所述第t个模板图像的特征图的更新特征；

结合第一个模板图像的特征图和所述更新特征，得到所述第t+1个模板图像；

其中，所述第一个模板图像是：根据第一帧图像中的所述目标的边界框对所述第一帧图像进行裁剪得到的包含所述目标的图像。

本申请实施例第二方面提供一种目标跟踪方法，所述方法包括：

基于第一方面任意技术方案提供的方法更新的模板图像；

根据更新后的模板图像对输入图像进行目标跟踪。

本申请实施例第三方面提供一种模板图像的更新装置，包括：

第一确定模块，用于根据第t+1帧图像确定所述第t+1帧图像的特征图，其中，t为正整数；

第一更新模块，用于根据所述第t+1帧图像的特征图，更新第t个模板图像得到第t+1个模板图像，以根据第t+1个模板图像对第t+2帧图像进行目标跟踪；其中，所述第t个模板图像，用于对所述第t+1帧图像进行目标跟踪；第t个模板图像和第t+1个模板图像中均包含要跟踪的目标。

基于上述方案，所述装置还包括：