[发明专利]用于跟踪视频中的目标图形的方法和装置

说明书

本申请实施例公开了用于跟踪视频中的目标图形的方法和装置。该方法的一具体实施方式包括：在目标视频的上一帧图像中确定目标图形的角点的位置信息和相邻的角点构成的第一边的参数信息；跟踪上一帧图像中的各角点获得当前帧图像中的目标图形的各角点的位置信息以确定第二边的参数信息；响应于确定第一边和第二边的相似度小于第一预设阈值，对各第一边的参数信息进行预测以获取预测的参数信息，并根据预测的参数信息生成多个候选目标图形；根据多个候选目标图形的图像特征确定目标图形在当前帧图像中的最终位置。该实施方式通过融合目标图形的角点和边的特征实现目标图形的追踪，并对追踪不准确的结果进行优化，提高了目标图形追踪的准确性。

技术领域

本申请实施例涉及计算机技术领域，具体涉及互联网技术领域，尤其涉及用于跟踪视频中的目标图形的方法和装置。

背景技术

在视频处理中，如果需要对某视频对象进行编辑，如水印添加、做马赛克掩膜、颜色校正等，都需要准确地知道其所在的位置。

现有技术提供了多种用于跟踪视频中待编辑对象的视频目标图形跟踪的方法，但是各方法通常仅适用于部分场景条件，很难适用于多种复杂场景的情况。例如，现有的基于稀疏光流/稠密光流的视频跟踪方法，在目标图形出现遮挡的情况下难以准确地确定目标图形；现有的基于特征点匹配的视频跟踪方法，不适用于纹理信息较弱的目标图形的跟踪；对于图像逐帧分割的方法，分割算法在复杂场景下无法准确地识别目标图形的边界。因此，如何在各种复杂场景下准确地跟踪视频的各帧图像中的目标图形成为亟待解决的问题。

发明内容

本申请实施例提出了用于跟踪视频中的目标图形的方法和装置。

第一方面，本申请实施例提供了一种用于跟踪视频中的目标图形的方法，该方法包括：在目标视频的与当前帧图像相邻的上一帧图像中确定目标图形的角点的位置信息和相邻的角点构成的第一边的参数信息，其中，第一边为上一帧图像中的目标图形的边；跟踪上一帧图像中的目标图形的各角点获得当前帧图像中的目标图形的各角点的位置信息，以确定第二边的参数信息，其中，第二边为当前帧图像中的目标图形的相邻的角点构成的边；响应于确定第一边和与该第一边对应的第二边的相似度小于第一预设阈值，对上一帧图像中的各第一边的参数信息进行预测以获取预测的参数信息，并根据预测的参数信息生成多个候选目标图形；根据多个候选目标图形的图像特征确定目标图形在当前帧图像中的最终位置。

在一些实施例中，在目标视频的与当前帧图像相邻的上一帧图像中确定目标图形的角点的位置信息和相邻的角点构成的第一边的参数信息，包括：在目标视频的上一帧图像中确定目标图形的各角点的位置信息；根据所确定的各角点的位置信息，确定上一帧图像中的目标图形的每条第一边的参数信息。

在一些实施例中，跟踪上一帧图像中的目标图形的各角点获得当前帧图像中的目标图形的各角点的位置信息，以确定第二边的参数信息，包括：采用KLT跟踪算法跟踪上一帧图像中的各角点，获得当前帧图像中的目标图形的各角点的位置信息；利用所获得的各角点的位置信息，确定当前帧图像中的目标图形的每条第二边的参数信息。

在一些实施例中，在跟踪上一帧图像中的目标图形的各角点获得当前帧图像中的目标图形的各角点的位置信息，以确定第二边的参数信息之后，方法还包括：对于上一帧图像中由任意两个相邻角点所确定的第一边和跟踪该两个角点获得的位置信息所确定的当前帧图像中的第二边，利用该第一边的参数信息和第二边的参数信息计算该第一边和第二边的相似度；响应于确定第一边和第二边的相似度大于或等于第一预设阈值，利用跟踪获得的各角点的位置信息确定目标图形在当前帧图像中的最终位置。

在一些实施例中，对上一帧图像中的各第一边的参数信息进行预测获取预测的参数信息，并根据所获取的预测的参数信息生成多个候选目标图形，包括：利用卡尔曼滤波器对上一帧图像中的各第一边的参数信息进行预测，获取预测的参数信息；利用预测的参数信息确定各预测边，根据各预测边确定多个预测角点；以预测角点为中心，在当前帧图像中确定每个预测角点的候选角点；利用各预测角点和各候选角点，生成多个候选目标图形。