一种高效的基于相似度的跨摄像头的目标重识别方法技术

本发明专利技术公开了一种高效的基于相似度的跨摄像头的目标重识别方法，该方法通过两组待匹配目标，得到多个匹配对及其相似度分数；针对双方均未匹配的匹配对，每次只取其中相似度分数较高的一部分匹配对，按照相似度分数从大到小的顺序遍历，并输出匹配对及其相似度分数，作为匹配结果；当某个匹配对中的任一待匹配目标已在匹配结果中出现，则不能作为匹配结果输出；重复遍历未匹配的匹配对直到匹配结果达到预期。本发明专利技术首先解决了基于相似度的多目标匹配问题，并相对于其他多目标匹配方法大幅度降低了时间复杂度，提升了效率。提升了效率。提升了效率。

【技术实现步骤摘要】
一种高效的基于相似度的跨摄像头的目标重识别方法


[0001]本专利技术属于多目标跟踪、目标重识别、多目标匹配
，尤其涉及一种高效的基于相似度的跨摄像头的目标重识别方法。

技术介绍

[0002]随着计算机视觉在人工智能领域的快速发展，在监控以及无人驾驶领域中，单摄像头的多目标跟踪以及跨摄像头的目标重识别算法变得越来越重要，在多目标跟踪以及目标重识别的整体框架中，多目标匹配算法是影响整体跟踪以及重识别算法效率的重要一环，提升多目标匹配算法的效率，即可大幅度提升多目标跟踪的帧率以及目标重识别的效率，对后续算法的实现影响很大。现阶段高效的且和匹配相似度相关的多目标匹配算法较少。

技术实现思路

[0003]本专利技术的目的在于针对现有技术的不足，提供一种高效的基于相似度的跨摄像头的目标重识别方法。
[0004]本专利技术的目的是通过以下技术方案来实现的：一种高效的基于相似度的跨摄像头的目标重识别方法，包括：通过两个摄像头拍摄到两张图片，分别对两张图片进行目标检测，得到对应的两组目标检测结果；其中，一组目标检测结果包括A个待匹配目标，另一组目标检测结果包括B个待匹配目标；分别对两组待匹配目标进行特征提取，得到每个待匹配目标的特征向量；不同组的待匹配目标两两配对，并根据特征向量计算相似度分数，得到A*B个匹配对及其相似度分数，相似度分数在[0,1]之间；针对双方均未匹配的匹配对，每次只取其中相似度分数较高的一部分匹配对，按照相似度分数从大到小的顺序遍历，并输出匹配对及其相似度分数，作为匹配结果；当某个匹配对中的任一待匹配目标已在匹配结果中出现，则不能作为匹配结果输出；如果当前匹配结果中的匹配对数量未达到预期的K组，则再次针对双方均未匹配的匹配对中相似度分数较高的部分，按照相似度分数从大到小的顺序遍历并输出，直到匹配结果达到预期；其中0＜K≤min(A,B)。
[0005]进一步地，目标检测结果包括行人。
[0006]进一步地，包括以下步骤：1)将A*B的相似度分数的二维矩阵，转换为一维向量M，其中M的长度为A*B，并记录每个相似度分数在原二维矩阵的行列值，对应在A和B的待匹配目标中索引号；2)维护一个结果容器R，存储已经选择的匹配对[i,j,N
ij
]，其中N
ij
为相似度分数，i为分数N
ij
在A中的索引号，j为分数N
ij
在B中的索引号；3)从向量M中随机选取一个相似度分数N；4)遍历向量M的相似度分数，比N大的分数放在左堆H_left，比N小的分数放在右堆
H_right；相似度分数N放在左堆或右堆；5)对左堆H_left进行排序，得到H_left_sorted；6)按照从大到小的顺序遍历堆H_left_sorted；若当前索引的i或者j不在结果容器R内，则得到匹配对[i,j,N
ij
]，加入结果容器R内；若当前索引的i或者j在结果容器R内，则跳过当前相似度分数继续遍历；7)若结果容器R内的个数大于等于K，则结束匹配，得到结果容器R；8)若结果容器R内的个数小于K，则遍历结果容器R，根据R中的索引过滤右堆H_right中具有相同的i或者相同的j的相似度分数，更新右堆H_right；将过滤后的右堆H_right作为新的向量M，重新执行步骤3)~8)直到结果容器R的匹配对大于等于K，得到结果容器R。
[0007]进一步地，设定匹配结果的相似度分数阈值，剔除匹配结果中低于阈值的匹配对。
[0008]进一步地，目标检测采用CenterNet目标检测算法。
[0009]进一步地，待匹配目标的特征向量通过分类器进行特征提取得到。
[0010]进一步地，所述分类器为ResNet50神经网络。
[0011]进一步地，采用两个待匹配目标的特征向量的余弦相似度作为相似度分数。
[0012]本专利技术的有益效果是：本专利技术在保证多目标匹配精度的同时，可大幅度降低多目标匹配算法的时间复杂度，从而提升多目标匹配算法的效率，即可大幅度提升多目标跟踪的帧率以及目标重识别的效率，对后续算法的实现影响很大。
附图说明
[0013]图1为本专利技术跨摄像头的目标重识别方法的流程示意图。
具体实施方式
[0014]本专利技术一种高效的基于相似度的跨摄像头的目标重识别方法，包括：输入两个摄像头拍摄到两张图片，分别为Data_1以及Data_2。
[0015]训练目标检测网络，分别对Data_1与Data_2数据进行目标检测，得到Data_1与Data_2的目标检测结果Detect_1以及Detect_2。
[0016]分别通过Detect_1与Detect_2将输入数据Data_1与Data_2的目标框选出来，得到Data_1与Data_2的不同目标数据Target_1_i以及Target_2_j，其中0<i<A，0<j<B，A、B分别为Data_1与Data_2目标检测的数量。
[0017]训练一个分类器，分别对Target_1_i与Target_2_j进行特征提取，得到A组与B组不同目标的特征向量F_1_i、F_2_j。
[0018]分别对特征向量F_1_i与F_2_j进行余弦相似度计算，得到两组待匹配目标的A*B个匹配对及其相似度分数，相似度分数在[0,1]之间；针对双方均未匹配的匹配对，每次只取其中相似度分数较高的一部分匹配对，按照相似度分数从大到小的顺序遍历，并输出匹配对及其相似度分数，作为匹配结果；当某个匹配对中的任一待匹配目标已在匹配结果中出现，则不能作为匹配结果输出；如果当前匹配结果中的匹配对数量未达到预期的K组，则再次针对双方均未匹配的匹配对中相似度分数较高的部分，按照相似度分数从大到小的顺序遍历并输出，直到匹
配结果达到预期；其中0＜K≤min(A,B)。
[0019]本专利技术通过多目标匹配结果，完成Target_1_i与Target_2_j的不同目标匹配，完成不同采集设备下不同目标的跟踪。
[0020]为了使本专利技术的目的、技术方案和技术效果更加清楚明白，以下结合说明书附图和实施例，对本专利技术作进一步详细说明。
[0021]针对于视频多目标跟踪以及跨摄像头的目标重识别任务，本专利技术可以高效地完成视频不同帧的多目标匹配以及不同摄像头之间的多目标匹配。
[0022]具体地，如图1所示，针对于跨摄像头采集到的图像数据的目标重识别任务：1）摄像头1拍摄的图片通过CenterNet目标检测算法检测出20个行人目标，摄像头2拍摄的图片通过CenterNet目标检测算法检测出30个行人目标。
[0023]2）采用ResNet50神经网络对摄像头1的20个目标和摄像头2的30个目标进行特征提取，分别得到不同目标的特征向量。
[0024]3）分别对20个目标的特征向量和30个目标的特征向量进行余弦相似度计算，得到20*30的相似度矩阵，分别表示两个摄像头不同目标之本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种高效的基于相似度的跨摄像头的目标重识别方法，其特征在于，包括：通过两个摄像头拍摄到两张图片，分别对两张图片进行目标检测，得到对应的两组目标检测结果；其中，一组目标检测结果包括A个待匹配目标，另一组目标检测结果包括B个待匹配目标；分别对两组待匹配目标进行特征提取，得到每个待匹配目标的特征向量；不同组的待匹配目标两两配对，并根据特征向量计算相似度分数，得到A*B个匹配对及其相似度分数，相似度分数在[0,1]之间；针对双方均未匹配的匹配对，每次只取其中相似度分数较高的一部分匹配对，按照相似度分数从大到小的顺序遍历，并输出匹配对及其相似度分数，作为匹配结果；当某个匹配对中的任一待匹配目标已在匹配结果中出现，则不能作为匹配结果输出；如果当前匹配结果中的匹配对数量未达到预期的K组，则再次针对双方均未匹配的匹配对中相似度分数较高的部分，按照相似度分数从大到小的顺序遍历并输出，直到匹配结果达到预期；其中0＜K≤min(A,B)。2.如权利要求1所述高效的基于相似度的跨摄像头的目标重识别方法，其特征在于，目标检测结果包括行人。3.如权利要求1所述高效的基于相似度的跨摄像头的目标重识别方法，其特征在于，包括以下步骤：1)将A*B的相似度分数的二维矩阵，转换为一维向量M，其中M的长度为A*B，并记录每个相似度分数在原二维矩阵的行列值，对应在A和B的待匹配目标中索引号；2)维护一个结果容器R，存储已经选择的匹配对[i,j,N
ij
]，其中N
ij
为相似度分数，i为分数N
ij
在A中的索引号，j为分数N
ij
在B中的索引号；3)从向量...

【专利技术属性】
技术研发人员：华炜，马也驰，张顺，
申请(专利权)人：之江实验室，
类型：发明
国别省市：