[发明专利]目标关联方法、装置和计算机可读存储介质

说明书

本公开涉及一种目标关联方法、装置和计算机可读存储介质，涉及视频处理技术领域。该方法包括：获取视频的待处理帧中的一个或多个目标；根据视频的多个待处理帧中各目标的候选关联关系，建立决策树，决策树的一个节点对应一个目标；计算决策树中各父节点与各父节点的子节点之间的关联程度；根据关联程度，筛选决策树中的分支；根据筛选结果，对各目标进行关联。本公开的技术方案能够提高目标关联的处理效率。

技术领域

本公开涉及视频处理技术领域，特别涉及一种目标关联方法、目标关联装置和计算机可读存储介质。

背景技术

目标跟踪技术多采用“检测+关联”的方案，即先从视频的每帧中将目标检测出来，再把这些目标检测框与身份关联起来。

相关技术中，目标关联处理主要采用如匈牙利算法(Hungarian method)等递归算法。

发明内容

本公开的发明人发现上述相关技术中存在如下问题：需要迭代求解，导致处理效率低下。

鉴于此，本公开提出了一种目标关联技术方案，能够提高目标关联的处理效率。

根据本公开的一些实施例，提供了一种目标关联方法，包括：获取视频的待处理帧中的一个或多个目标；根据所述视频的多个待处理帧中各目标的候选关联关系，建立决策树，所述决策树的一个节点对应一个所述目标；计算所述决策树中各父节点与所述各父节点的子节点之间的关联程度；根据所述关联程度，筛选所述决策树中的分支；根据筛选结果，对所述各目标进行关联。

在一些实施例中，所述决策树的一个层包含一个所述待处理帧中的一个或多个所述目标，所述决策树的一个分支代表一种候选关联关系。

在一些实施例中，将所述关联程度高于阈值的父节点和所述父节点的子节点确定为关联节点对；删除所述决策树中不符合各所述关联节点对的关联关系的分支；将所述决策树中剩余的分支作为所述筛选结果。

在一些实施例中，按照所述关联程度从高到低的顺序，读取所述关联节点对；删除所述决策树中不符合当前读取的所述关联节点对的关联关系的分支，直到读取完所有的所述关联节点对。

在一些实施例中，所述多个待处理帧为所述视频的多个相邻帧；所述根据多个待处理帧中各目标的候选关联关系建立决策树包括：按照所述多个待处理帧的时间顺序，建立所述决策树。

在一些实施例中，该方法还包括：提取各所述目标的特征向量；根据所述特征向量之间的距离计算所述关联程度。

在一些实施例中，所述候选关联关系根据所述待处理帧中的目标位置与所述待处理帧的相邻帧中的目标位置建立。

根据本公开的另一些实施例，提供一种目标关联装置，包括：获取单元，用于获取视频的待处理帧中的一个或多个目标；建立单元，用于根据所述视频的多个待处理帧中各目标的候选关联关系建立决策树，所述决策树的一个节点对应一个所述目标；计算单元，用于计算所述决策树中各父节点与所述各父节点的子节点之间的关联程度；筛选单元，用于将根据所述关联程度，筛选所述决策树中的分支；关联单元，用于根据筛选结果，对所述各目标进行关联。

在一些实施例中，所述决策树的一个层包含一个所述待处理帧中的一个或多个所述目标，所述决策树的一个分支代表一种候选关联关系。

在一些实施例中，所述筛选单元将所述关联程度高于阈值的父节点和所述父节点的子节点确定为关联节点对，删除所述决策树中不符合各所述关联节点对的关联关系的分支，将所述决策树中剩余的分支作为所述筛选结果。

在一些实施例中，所述筛选单元按照所述关联程度从高到低的顺序，读取所述关联节点对，删除所述决策树中不符合当前读取的所述关联节点对的关联关系的分支，直到读取完所有的所述关联节点对。