[发明专利]目标跟踪方法、装置、电子设备及存储介质

说明书

本申请提供了一种目标跟踪方法、装置、电子设备及存储介质，所述方法包括：获取用户在当前图像帧中选定的目标位置，并以所述目标位置为中心生成所述用户选定的第一目标框；基于检测算法在所述当前图像帧内进行检测，得到检测框；计算所述检测框与所述第一目标框之间的重叠度，并将所述重叠度最高的检测框作为目标所在的第二目标框；基于所述第二目标框，确定所述当前图像帧中的目标所在区域。本申请实施例能够提高目标跟踪的效率。

技术领域

本申请涉及图像处理领域，具体涉及一种目标跟踪方法、装置、电子设备及存储介质。

背景技术

随着图像处理技术的高速发展，在众多领域的应用中，均需要对图像中的目标进行跟踪。尤其是在无人机飞行过程中进行目标跟踪时，一般来说，飞手会先选定要跟踪的目标，进而用户终端再根据飞手选定的目标进行目标跟踪。

由于各类因素的影响，飞手往往需要经过多次选定才能正确选定到目标。一旦飞手选定的目标与实际目标之间存在较大的误差，便会导致在后续图像帧中无法成功跟踪目标，跟踪效率较低。

发明内容

本申请的一个目的在于提出一种目标跟踪方法、装置、电子设备及存储介质，能够提高目标跟踪的效率。

根据本申请实施例的一方面，公开了一种目标跟踪方法，所述方法包括：

获取用户在当前图像帧中选定的目标位置，并以所述目标位置为中心生成所述用户选定的第一目标框；

基于检测算法在所述当前图像帧内进行检测，得到检测框；

计算所述检测框与所述第一目标框之间的重叠度，并将所述重叠度最高的检测框作为目标所在的第二目标框；

基于所述第二目标框，确定所述当前图像帧中的目标所在区域。

根据本申请实施例的一方面，公开了一种目标跟踪装置，所述装置包括：

获取模块，配置为获取用户在当前图像帧中选定的目标位置，并以所述目标位置为中心生成所述用户选定的第一目标框；

检测模块，配置为基于检测算法在所述当前图像帧内进行检测，得到检测框；

计算模块，配置为计算所述检测框与所述第一目标框之间的重叠度，并将所述重叠度最高的检测框作为目标所在的第二目标框；

确定模块，配置为基于所述第二目标框，确定所述当前图像帧中的目标所在区域。

在本申请的一示例性实施例中，所述获取模块配置为：

获取所述用户针对所述第一目标框调节的波门；

基于所述波门计算得到所述第一目标框的第一尺寸参数，并将以所述目标位置为中心生成的所述第一尺寸参数的框作为所述第一目标框。

在本申请的一示例性实施例中，所述获取模块配置为：

获取针对所述第一尺寸参数中的轮廓线长度预设的最大值与针对所述第一尺寸参数中的轮廓线长度预设的最小值之间的差值；

基于所述差值与针对所述波门预设的最大值之间的比值，对所述波门进行修正，得到修正值；

基于所述修正值与所述最小值的和，获取所述第一尺寸参数中的轮廓线长度。

在本申请的一示例性实施例中，所述检测模块配置为：

基于检测算法在包含所述当前图像帧在内的与所述当前图像帧相邻的预设数目个的图像帧进行检测，得到检测框。

在本申请的一示例性实施例中，所述计算模块配置为：

计算所述检测框与所述第一目标框之间的交集面积；