[发明专利]一种目标搜索方法及装置

说明书

本发明公开了一种目标搜索方法及装置。所述方法包括：获取目标视频帧中各目标的特征表达，基于各目标的特征表达，构建以待搜索目标为中心节点，其他目标为上下文节点且由上下文节点指向中心节点的源图。获取候选视频帧中各候选目标的特征表达，基于各候选目标的特征表达和各目标的特征表达，确定与各目标对应的候选目标，构建以与待搜索目标对应的候选目标为中心节点，与其他目标对应的候选目标为上下文节点且由上下文节点指向中心节点的目标图。基于孪生残差图卷积神经网络获得源图和目标图的图嵌入向量。至少基于源图的图嵌入向量和目标图的图嵌入向量确定候选视频帧中的待搜索目标。本发明的方案，提高了目标搜索的准确度且具有较强的鲁棒性。

技术领域

本发明涉及计算机视觉技术领域，特别涉及一种目标搜索方法及装置、计算机设备、计算机可读存储介质。

背景技术

一般来讲，目标搜索可定义为在海量的监控视频帧中搜索特定目标的过程。目标搜索技术是一种自动化的目标检测识别技术，其可以在监控网络中快速定位到感兴趣的目标。目标搜索技术联合了目标检测和目标重识别，其作为智能视频监控中的一项关键技术，在近些年获得了计算机视觉领域的广泛关注。以目标为行人为例，行人搜索在安防寻人、人体行为分析等应用中具有重要的应用价值。

目前通常根据待搜索目标和候选目标的特征来计算二者之间的个体相似度，然后根据二者之间的个体相似度来对候选目标进行排序，以实现对待搜索目标的搜索。然而在监控视频中，目标通常会因为视角、光照等变化，以及遮挡的问题而发生巨大的变化。如对于监控视频中的行人而言，会因为姿态、视角、光照、是否被遮挡等原因，其外观在不同视频帧中会呈现较大的变化。然而采用目前的目标搜索方法，搜索的准确度较低。

因此，如何能够在视频中搜索出目标，提高目标搜索的准确度，成为目前亟待解决的问题之一。

发明内容

本发明提供一种目标搜索方法、装置、计算机设备及计算机可读存储介质，以在不同的视频帧中准确的搜索出目标。

本发明提供一种目标搜索方法，包括：

获取目标视频帧中各目标的特征表达，基于各目标的特征表达，构建以待搜索目标为中心节点，其他目标为上下文节点且由上下文节点指向中心节点的源图；

获取候选视频帧中各候选目标的特征表达，基于各候选目标的特征表达和各目标的特征表达，确定与各目标对应的候选目标，构建以与待搜索目标对应的候选目标为中心节点，与其他目标对应的候选目标为上下文节点且由上下文节点指向中心节点的目标图，其中，与各目标对应的候选目标是指与每一个目标最相似的一候选目标；

基于孪生残差图卷积神经网络获得源图的图嵌入向量和目标图的图嵌入向量；

至少基于所述源图的图嵌入向量和所述目标图的图嵌入向量确定所述候选视频帧中的待搜索目标。

可选的，所述至少基于所述源图的图嵌入向量和所述目标图的图嵌入向量确定所述候选视频帧中的待搜索目标包括：

基于所述源图的图嵌入向量和所述目标图的图嵌入向量计算待搜索目标和与其对应的候选目标的图相似度；

基于待搜索目标和与其对应的候选目标的图相似度确定所述候选视频帧中的待搜索目标。

可选的，所述至少基于所述源图的图嵌入向量和所述目标图的图嵌入向量确定所述候选视频帧中的待搜索目标包括：

基于所述源图的图嵌入向量和所述目标图的图嵌入向量计算待搜索目标和与其对应的候选目标的图相似度；

基于所述待搜索目标和与其对应的候选目标的图相似度对所述待搜索目标和与其对应的候选目标的相似度进行修正以获得修正后的待搜索目标和与其对应的候选目标的相似度；

基于修正后的待搜索目标和与其对应的候选目标的相似度确定所述候选视频帧中的待搜索目标。