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

说明书

本公开实施例公开了一种目标跟踪方法、装置、电子设备以及可读存储介质。该目标跟踪方法包括：获取上一帧的目标边界框；将上一帧的目标边界框的尺寸放大以生成搜索区域；将当前帧在搜索区域内的图像数据缩放到预定尺寸；将缩放后的图像数据输入目标跟踪网络，以便从多个候选锚点框中确定目标锚点框，并获取目标锚点框的位置偏移量和尺寸调整信息；基于位置偏移量和尺寸调整信息调整目标锚点框，以便从搜索区域中确定目标区域；基于搜索区域以及目标区域在搜索区域中的位置，确定目标在当前帧中的位置，表示为当前帧的目标边界框，从而减小了跳帧处理下目标在图像上的变化对目标跟踪效果的影响，提高了目标跟踪的正确性。

技术领域

本公开涉及视频技术领域，具体涉及一种目标跟踪方法、装置、电子设备以及可读存储介质。

背景技术

目标跟踪是计算机视觉研究领域的热点之一，并得到广泛应用。相机的跟踪对焦、无人机的自动目标跟踪等都需要用到了目标跟踪技术。另外还有特定物体的跟踪，比如人体跟踪，交通监控系统中的车辆跟踪，人脸跟踪和智能交互系统中的手势跟踪等。简单来说，目标跟踪就是在连续的视频序列中，建立所要跟踪物体的位置关系，得到物体完整的运动轨迹。给定图像第一帧的目标坐标位置，计算在下一帧图像中目标的确切位置。

跳帧是指按一定间隔从视频中抽取部分视频帧进行处理。本发明人发现，在运动的过程中，目标可能会呈现一些图像上的变化，比如姿态、形状或尺度的变化，尤其是在跳帧的情况下，这种变化对目标跟踪的效果产生严重的影响。

发明内容

为了解决相关技术中的问题，本公开实施例提供一种目标跟踪方法、装置、电子设备以及可读存储介质。

第一方面，本公开实施例中提供了一种目标跟踪方法。

具体地，所述目标跟踪方法包括：

获取上一帧的目标边界框；

将所述上一帧的目标边界框的尺寸放大以生成搜索区域；

将当前帧在所述搜索区域内的图像数据缩放到预定尺寸；

将缩放后的图像数据输入目标跟踪网络，以便从多个候选锚点框中确定目标锚点框，并获取所述目标锚点框的位置偏移量和尺寸调整信息；

基于所述位置偏移量和尺寸调整信息调整所述目标锚点框，以便从所述搜索区域中确定目标区域；

基于所述搜索区域以及所述目标区域在搜索区域中的位置，确定目标在所述当前帧中的位置，表示为所述当前帧的目标边界框。

结合第一方面，本公开在第一方面的第一种实现方式中，所述将缩放后的图像数据输入目标跟踪网络，以便从多个候选锚点框中确定目标锚点框，并获取所述目标锚点框的位置偏移量和尺寸调整信息包括：

将缩放后的图像数据输入目标跟踪网络，以获取目标置信度、位置偏移量和尺寸调整信息；

基于所述目标置信度，从多个候选锚点框中确定目标锚点框。

结合第一方面的第一种实现方式，本公开在第一方面的第二种实现方式中，所述目标跟踪网络包括五个串联的卷积层，在第一个卷积层和第二个卷积层后分别设置有池化层，最后一个卷积层包括六个通道，

其中，所述六个通道中的两个通道用于输出所述目标置信度，所述六个通道中的另外四个通道用于输出所述位置偏移量和尺寸调整信息，

其中，所述第一个卷积层的输入尺寸与所述预定尺寸一致，所述最后一个卷积的输出尺寸与所述候选锚点框的数量相匹配。

结合第一方面的第一种实现方式，本公开在第一方面的第三种实现方式中，所述目标置信度包括每个候选锚点框为目标锚点框的第一概率和非目标锚点框的第二概率，所述基于所述目标置信度，从多个候选锚点框中确定目标锚点框包括：