匈牙利算法相关介绍

2024-03-02 16:20
文章标签 算法 介绍 相关 匈牙利

本文主要是介绍匈牙利算法相关介绍,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

重要说明:本文从网上资料整理而来,仅记录博主学习相关知识点的过程,侵删

一、参考资料

匈牙利算法匹配问题?
Exactly how the Hungarian Algorithm works
多目标跟踪数据关联之匈牙利算法
五分钟小知识:什么是匈牙利算法
论文:The Hungarian Method for the Assignment Problem

二、相关介绍

1. 二分图

1.1 二分图的概念

二分图(Bipartite Graph)又称作二部图,是图论中的一种特殊模型。二分图的顶点集可以被划分为两个互不相交的独立子集 U 和 V :
G = ( U , V , E ) G=(U,V,E) G=(U,V,E)
二分图边集合 E 中的每一条边分别连接了顶点子集 U 和 V 中的一个顶点,但 U 和 V 内部的顶点互不相连。一个典型的二分图如下所示:
在这里插入图片描述

简单来说,如果图中所有顶点可以被分为两个集合,图中所有的边的头和尾不属于同一个顶点集合,而是跨越两个集合,则这个图是一个二分图。

1.2 无向图生成二分图

无向图如下所示:
在这里插入图片描述

将顶点a,b,c,d作为集合A,将e,f,g,h作为集合B,转换为二分图:
在这里插入图片描述

可以看出,图中顶点可以划分为A,B两个集合,而任意一条边的头和尾又分别隶属于集合A和集合B,因此,此图为二分图。

1.3 判定二分图

判定一个图是否是二分图等价于判定该图是否是二色图,即图中所有顶点是否能被染色成两类颜色(需要保证相连顶点不能同色),可以通过广度优先搜索(Breadth First Search,BFS)实现。

C++代码示例,使用邻接表来存储图,时间复杂度为O(V+E)。

#include <iostream>
#include <vector>
#include <queue>
#include <utility>/*** @brief 用于判断输入的图是否是二分图** @param V 图的顶点数量* @param adj 图的邻接表表示* @return bool 图是否是二分图:true - 是,false - 不是*/
bool IsBipartite(int V, std::vector<int> adj[])
{// 存储所有顶点颜色的 vector 容器,初值 -1 表示未染色,0 表示红色,1 表示蓝色std::vector<int> col(V, -1);// 用于 BFS 过程的 FIFO 队列,元素类型是顶点索引及其颜色组成的二元组std::queue<std::pair<int, int>> q;// 遍历所有顶点for (int i = 0; i < V; i++){// 顶点 i 尚未染色if (col[i] == -1){// 将顶点 i 染成红色 0,并将其压入 BFS FIFO 队列col[i] = 0;q.push({ i, 0 });// 处理 BFS FIFO 队列中的各顶点while (!q.empty()){auto p = q.front();q.pop();// 当前顶点int v = p.first;// 当前顶点的颜色int c = p.second;// 遍历当前顶点的所有相连顶点for (int j : adj[v]){// 若相连顶点 j 与当前顶点颜色相同,则输入的图不是二分图if (col[j] == c) return false;// 若相连顶点 j 尚未染色if (col[j] == -1){// 将相连顶点 j 染成与当前顶点颜色相反的颜色,并将其压入 BFS// FIFO 队列col[j] = 1 - c;q.push({ j, col[j] });}}}}}return true;
}int main()
{int V, E;V = 4 , E = 8;// 使用邻接表存储图std::vector<int> adj[V];adj[0] = {1, 3};adj[1] = {0, 2};adj[2] = {1, 3};adj[3] = {0, 2};IsBipartite(V, adj) ? std::cout << "输入的图是二分图" << std::endl: std::cout << "输入的图不是二分图" << std::endl;return 0;
}

2. 二分图匹配

2.1 匹配

在图论中,一个匹配(matching)是指一个边的集合,其中任意两条边都没有公共顶点。

如下图所示,寻找该二分图的一个匹配。
在这里插入图片描述

下图中,红色的边为一个匹配。
在这里插入图片描述

2.2 匹配点与匹配边

给定图 G = ( V , E ) G=(V,E) G=(V,E),它的一个匹配(Matching) M \text{M} M 表示包含于边集合 E \text{E} E 的一个子集,即M 由 E \text{E} E 中的若干条边组成
M ⊆ E M\subseteq E ME
匹配中的任意两条边之间没有公共顶点。下图中的两条红色边构成了它的一个匹配:
在这里插入图片描述

匹配 M \text{M} M 中的边被称为匹配边,上图中的两条红色边都是匹配边;边集合 E \text{E} E 中不属于匹配 M \text{M} M 的边被称为未匹配边,上图中的灰色边都是未匹配边。

匹配边的端点被称为匹配点,上图中顶点 u 4 、 u 5 、 v 1 、 v 3 u_4\text{、}u_5\text{、}v_1\text{、}v_3 u4u5v1v3 都是匹配点;顶点集合 U \text U U V \text V V 中不是匹配边端点的其它顶点被称为未匹配点,上图中顶点 u 1 , u 2 , u 3 , v 2 , v 4 u_1,u_2,u_3,v_2,v_4 u1,u2,u3,v2,v4 都是未匹配点。

2.3 最大匹配

一个图所有匹配中,所含匹配边数最多的匹配,称为这个图的最大匹配(Maximum-Cardinality Matching)。最大匹配不唯一,如下图所示,下图的两个匹配都是最大匹配,最大匹配的边数为 4。
在这里插入图片描述

2.4 完美匹配

如果一个图的某个匹配中,所有的顶点都是匹配点,那么它就是一个完美匹配。完美匹配一定是最大匹配(完美匹配的任何一个点都已经匹配,添加一条新的匹配边一定会与已有的匹配边冲突),但并非每个图都存在完美匹配。

2.5 最大权匹配与最小权匹配

最大权匹配(Maximum-Weight Matching)表示的是有权图的所有匹配中边的权重之和最大的那些匹配,最小权匹配(Minimum-Weight Matching)表示的是有权图的所有匹配中边的权重之和最小的那些匹配。

对二分图所有的边赋上权重,将其转成有权二分图,如下图所示:
在这里插入图片描述

对于上面的有权二分图,我们将任意边上的权重记作 w \text{w} w,任意匹配记作 M \text{M} M,匹配 M \text{M} M 包含的所有匹配边的权重和记作函数 f ( M ) f(M) f(M),则有权二分图的最大权匹配(Maximum-Weight Bipartite Matching)问题和有权二分图的最小权匹配(Minimum-Weight Bipartite Matching)问题可以用如下数学语言进行描述:
{ f ( M ) = ∑ ( u , v ) ∈ M w u , v max ⁡ / min ⁡ f ( M ) \left\{\begin{array}{l}f(M)=\sum_{(u,v)\in M}w_{u,v}\\\max/\min f(M)\end{array}\right. {f(M)=(u,v)Mwu,vmax/minf(M)
多目标跟踪数据关联问题可以转化为有权二分图最小权匹配问题,跟踪过程中的图像帧可以分别看作是有权二分图中的顶点集合 U \text U U V \text V V,边的权重可以看作是前后帧目标之间的距离,通过某种方式计算得到的匹配距离(例如欧式距离),这个匹配距离称之为代价(Cost),所有的匹配距离构成了代价矩阵(Cost Matrix)。我们需要做的是,找到图像帧的匹配关系,使得总的匹配距离最小,代价最低。

3. 交替路径

给定图 G = ( V , E ) G=(V,E) G=(V,E) 和它的一个匹配 M \text M M,交替路(Alternating Path)描述的是图中的这样一条路径:从图中的某个未匹配点出发,交替经过未匹配边和匹配边形成的路径。下面的二分图中,路径 u 3 → v 1 → u 1 → v 3 → u 2 u_3\to v_1\to u_1\to v_3\to u_2 u3v1u1v3u2 就是一条交替路:
在这里插入图片描述

4. 增广路径

4.1 增广路径的概念

从图中的某个未匹配点起始,交替经过未匹配边和匹配边,并终止于不同于起始点的另一个未匹配点,则这条路径称为增广路径(Augmenting Path),增广路径是一条特殊的交替路。下面的二分图中,路径 u 1 → v 1 → u 5 → v 4 u_1\to v_1\to u_5\to v_4 u1v1u5v4 就是一条增广路径。
在这里插入图片描述

4.2 增广路的性质

给定图 G = ( V , E ) G=(V,E) G=(V,E) 和它的一个匹配 M \text M M 以及增广路 P \text P P,并将 P \text P P 上所有的边记作集合 E P \text E_P EP,则有如下三个非常重要的性质:

  1. Berge 定理:对于给定的图 G \text G G 和它的一个匹配 M \text M M M \text M M G \text G G 的最大匹配的充要条件是 G \text G G 中不存在匹配 M \text M M 的增广路;

  2. E P \text E_P EP 中边的数量一定为奇数,且增广路 P \text P P 的第奇数条边不属于匹配 M \text M M,第偶数条边属于匹配 M \text M M,这意味着, E P \text E_P EP 中的未匹配边数量一定比匹配边数量多 1;

  3. 通过性质 1 不难发现,通过将 E P \text E_P EP 中的未匹配边取反变成匹配边,匹配边取反变成未匹配边,就可以多出 1 条匹配边,取反得到的新匹配边和 M \text M M 中不属于 E P \text E_P EP 的剩余边可以构成一个更大的匹配 M ′ M^{\prime} M。在集合论中, M ′ M^{\prime} M 被称作 M \text M M E P \text E_P EP 的对称差(Symmetric Difference),记作 M ⊖ E P M\ominus E_{P} MEP
    M ⊖ E P = ( M ∪ E P ) − ( M ∩ E P ) M\ominus E_P=(M\cup E_P)-(M\cap E_P) MEP=(MEP)(MEP)
    结合性质 1 和性质 3 我们不难发现,对于一个给定的二分图 G = ( U , V , E ) G=(U,V,E) G=(U,V,E) 和初始为空的匹配 M \text M M,我们只要反复搜索增广路就能逐渐扩展匹配的大小,最终当我们找不到增广路时就得到了一个最大匹配,下图中的示例很直观地展示了这个过程:
    在这里插入图片描述

最后得到的 M = { e 1 , 3 , e 3 , 1 , e 4 , 2 , e 5 , 4 } M=\{e_{1,3},e_{3,1},e_{4,2},e_{5,4}\} M={e1,3,e3,1,e4,2,e5,4} 就是输入二分图 G \text G G 的一个最大匹配。

三、匈牙利算法

匈牙利算法(Hungarian algorithm),是由匈牙利数学家Edmonds于1965年提出,因而得名。匈牙利算法是全局最近邻(Global Nearest Neighbor,GNN)数据关联思想的一种具体实现,其最早用于求解经济学领域中的任务分配问题,后来发展成为图论领域中求解有权二分图最小权匹配问题的一般性算法。

匈牙利算法本质是图算法,该算法的核心就是寻找增广路径,它是一种利用增广路径求二分图最大匹配的算法,是一种能够在多项式时间内解决分配问题的组合优化算法

1. 问题引入

在目标跟踪中,不同时刻物体的位置等信息发生变化,但我们需要让它保持同一个ID。

以目标跟踪小车为例,假设Frame1视频帧中出现一辆车,采用目标检测方法,可以一直把该黄车标成ID=1。
在这里插入图片描述

如果Frame2视频帧中出现了另外一台车,采用目标检测器只有分类出“车”的能力,无法知道黄车的ID是否等于1。
在这里插入图片描述

采用匈牙利算法,可以判断不同的ID。匈牙利算法会建立一个图,其中有当前帧t=1和前一帧t=0的节点,计算两帧节点之间的距离,距离越小,则当前帧的物体与前一帧物体相同的概率越大。
在这里插入图片描述

2. 适用场景

匈牙利算法多用于指派问题中,例如任务匹配问题。通过转化为二分图的形式,求解最大匹配,保证实现最优分配。

3. 匈牙利算法思想

**基本思想:**通过寻找增广路径,把增广路径中的匹配边和非匹配边的相互交换,这样就会多出一条匹配边,直到找不到增广路径为止。

以一个简单的例子,介绍匈牙利算法的思想。

看这么一个例子,把左边1,2,3,4和右边a,b,c,d来进行匹配。
在这里插入图片描述

一开始我们给1分配a,1和a之间连上红线表示建立匹配。
在这里插入图片描述

然后接着给2分配b,2和b连上红线表示匹配。
在这里插入图片描述

紧接着给3分配,这时候发现a,b已经都有所属了,我们尝试给1重新分配,把原来的分配拆掉,用蓝线表示。
在这里插入图片描述

但是很快我们发现1重新分配不了,b已经有所属,那么继续尝试给2重新分配,把原来的分配拆掉,用蓝线表示。2重新分配到c,用红线表示。
在这里插入图片描述

这个时候,1可以重新分配到b,用红线表示。
在这里插入图片描述

最后,3就可以分配到a,用红线表示。
在这里插入图片描述

对于4,由于c已经被分配,而且尝试给其他1,2,3重新分配无法实现,就此结束。

总结:匈牙利算法的基本原则就是在原有匹配基础上重新分配,看是否可以添加一个新的匹配。

4. 匈牙利算法流程

匈牙利算法就是用来解决分配问题(Assignment Problem),得到最优分配的结果。匈牙利算法的分配问题分 5 步工作:首先对矩阵进行缩减,然后交叉 0,最后再次缩减,直到我们可以对元素进行配对。总体流程如下:
在这里插入图片描述

本章以一个抽象例子,介绍匈牙利算法解决分配问题的流程。假设我们有两帧图片,第一帧图片中有CarA, CarB, CarC, 并且通过目标检测器找到了这三辆车;第二帧,同样,也看到了三辆车,CarD, CarE, CarF,两帧都只存在这三辆车,但由于目标检测器只能分辨出他们是车,而不能分辨出CarD对应的是CarA还是其它车。可以测量出两帧之间车的对应距离,绘制出代价矩阵(cost matrix)。例如,第二帧的CarF与第一帧的CarA距离为9m,如下图数字所示。
在这里插入图片描述

4.1 第一步:每行减去最小值

第一行最小的是9,则减去9,以此类推。最后变为:
在这里插入图片描述

4.2 第二步:每列减去最小值

第一列最小的是1,则减去1,以此类推。最后变为:
在这里插入图片描述

4.3 第三步:以最少数量的线条划掉所有零

划掉所有零有不少方式,如下图:
在这里插入图片描述

但明显最少的方式应该是两条:
在这里插入图片描述

如果这个数量大于等于矩阵的行列数,那么跳到第五步。

4.4 第四步:在剩下的矩阵中,减去最小值

在剩下的矩阵中,减去最小值;如果有零被交叉,那么把这个最小值加上去。

剩下的矩阵为【【3, 7】,【2, 5】】,都减去2。然后右叉的0加上2。变为:
在这里插入图片描述

然后重复第三步,最后变为:
在这里插入图片描述

这时我们有3条线了,那么跳到第五步。

4.5 第五步:从只有一个零的行开始一一对应,对应完则整个行列删除

第二行只有一个零,那么Car F 对应了car B,然后删掉行列。最后变为:
在这里插入图片描述

第三行,Car C对应了Car D。
在这里插入图片描述

剩下就是Car A对应Car E了。最终结果为:
在这里插入图片描述

5. Python实现匈牙利算法

scipy的库linear_sum_assignment

sklearn里的 linear_assignment() 函数以及scipy里的 linear_sum_assignment() 函数都实现了匈牙利算法,两者的返回值的形式不同:

import numpy as np 
from sklearn.utils.linear_assignment_ import linear_assignment
from scipy.optimize import linear_sum_assignmentcost_matrix = np.array([[10,15,9],[9,18,5],[6,14,3]
])matches = linear_assignment(cost_matrix)
print('sklearn API result:\n', matches)
matches = linear_sum_assignment(cost_matrix)
print('scipy API result:\n', matches)"""Outputs
sklearn API result:[[0 1][1 2][2 0]]
scipy API result:(array([0, 1, 2], dtype=int64), array([1, 2, 0], dtype=int64))
"""

在这里插入图片描述

可以看到,他直接帮你匹配好了,0(Car A) → 1(Car E)。

6. 匈牙利算法特例

NxN vs NxM

如果有 NxM 矩阵,则可以通过添加具有最大值的列来使其成为 NxN 矩阵。

刚刚我们的矩阵的行列数是相等的,那是因为两帧的图片都只出现了三辆车,那么如果第一帧三辆车,第二帧出现了四辆车呢? 如果出现这种情况,我们只需要添加一个新的边缘给我们的图标,这个值是我们原先的最大值就可以了。
在这里插入图片描述

最大值 vs 最小值

如果我们使用IOU匹配的话,我们需要优化的是最大值。我们只需要在第一步进行一个转换,如下所示:
在这里插入图片描述

例如,在Detenction 4, Tracking 1中(行4列1),cost metric 只有10,如果是IOU匹配的话,表示这个匹配度很低,我们把所有的值都让最大值减去这个值,比如10,我们让90 - 10 = 80。

四、目标跟踪算法

目前主流的目标跟踪算法,都是基于Tracking-by-Detection策略,即基于目标检测的结果来进行目标跟踪。DeepSort就是运用这个策略。

视频中不同时刻的同一个人,位置发生了变化,如何关联上呢?答案就是匈牙利算法和卡尔曼滤波。

  • 匈牙利算法:匹配当前帧的某个目标与前一帧的某个目标;
  • 卡尔曼滤波:基于目标前一时刻的位置,来预测当前时刻的位置,可以比目标检测器更准确的估计目标的位置。

1. 目标框的距离计算

目前经典的MOT多目标跟踪算法是sort、deepsort,其中sort算法是先经过匈牙利算法进行最大匹配,再经过卡尔曼滤波进行预测。

在实际的目标跟踪当中,我们获取前一帧的跟踪框track 和当前帧的检测框detection ,之间的距离Cost Metric 如何计算的呢?

检测框则是当前帧的目标检测器检测出来的bounding box, 跟踪框是上一帧最后输出预测出来的框,并不是检测框。

目标框的距离计算,常用有三种方法:欧式距离,IOU匹配,外观相似度(Convolutional Cost)。

1.1 欧式距离

我们可以把两帧的目标框的中心点进行一个距离的计算:
d = ( x 2 − x 1 ) 2 + ( y 2 − y 1 ) 2 \mathrm{d}=\sqrt{(x_2-x_1)_2+(y_2-y_1)^2} d=(x2x1)2+(y2y1)2
在这里插入图片描述

这种方法最简单,但如果目标的形状发生变化,或者目标与其他发生了重叠,会有不少问题。

1.2 IOU匹配

在这里插入图片描述

由图可知,这种方法有个问题,就是我们现在不是求最小,而是求最大,所以会与上述说的方法有一些出入。

1.3 外观相似度(Convolutional Cost)

IOU匹配是很常用的,但是如果出现了目标的重叠阻挡等问题,目标框也会发生变化,IOU匹配也会不准确。

但即使目标有遮挡,但如果我们一样能检测其中一部分,那么我们能不能用一个卷积网络提取其中的特征,然后和原本的框的特征进行相比呢?如下图:
在这里插入图片描述

加入了外观信息,借用ReID领域模型来提取特征,减少了ID switch,这也是DeepSort相比Sort的其中一个创新点。

五、DeepSort算法

目标跟踪初探(DeepSORT)

deep_sort_pytorch

1. DeepSort算法简介

在DeepSort算法中,匈牙利算法用来将前一帧中的跟踪框tracks与当前帧中的检测框detections进行关联匹配,通过外观信息(appearance information)和欧式距离或者IOU来计算代价矩阵。

源码解读

#  linear_assignment.py
def min_cost_matching(distance_metric, max_distance, tracks, detections, track_indices=None, detection_indices=None):...# 计算代价矩阵cost_matrix = distance_metric(tracks, detections, track_indices, detection_indices)cost_matrix[cost_matrix > max_distance] = max_distance + 1e-5# 执行匈牙利算法,得到匹配成功的索引对,行索引为tracks的索引,列索引为detections的索引row_indices, col_indices = linear_assignment(cost_matrix)matches, unmatched_tracks, unmatched_detections = [], [], []# 找出未匹配的detectionsfor col, detection_idx in enumerate(detection_indices):if col not in col_indices:unmatched_detections.append(detection_idx)# 找出未匹配的tracksfor row, track_idx in enumerate(track_indices):if row not in row_indices:unmatched_tracks.append(track_idx)# 遍历匹配的(track, detection)索引对for row, col in zip(row_indices, col_indices):track_idx = track_indices[row]detection_idx = detection_indices[col]# 如果相应的cost大于阈值max_distance,也视为未匹配成功if cost_matrix[row, col] > max_distance:unmatched_tracks.append(track_idx)unmatched_detections.append(detection_idx)else:matches.append((track_idx, detection_idx))return matches, unmatched_tracks, unmatched_detections

2. DeepSort算法流程

DeepSORT对每一帧的处理流程如下:

检测器得到bbox → 生成detections → 卡尔曼滤波预测→ 使用匈牙利算法将预测后的tracks和当前帧中的detecions进行匹配(级联匹配和IOU匹配) → 卡尔曼滤波更新。

**Frame 0:**检测器检测到了3个detections,当前没有任何tracks,将这3个detections初始化为tracks。
**Frame 1:**检测器又检测到了3个detections,对于Frame 0中的tracks,先进行预测得到新的tracks,然后使用匈牙利算法将新的tracks与detections进行匹配,得到(track, detection)匹配对,最后用每对中的detection更新对应的track。

2.1 检测器得到bbox

使用Yolo作为检测器,检测当前帧中的bbox:

#  demo_yolo3_deepsort.py
def detect(self):while self.vdo.grab():...bbox_xcycwh, cls_conf, cls_ids = self.yolo3(im)  # 检测到的bbox[cx,cy,w,h],置信度,类别idif bbox_xcycwh is not None:# 筛选出人的类别mask = cls_ids == 0bbox_xcycwh = bbox_xcycwh[mask]bbox_xcycwh[:, 3:] *= 1.2cls_conf = cls_conf[mask]...

2.2 生成detections

将检测到的bbox转换成detections:

#  deep_sort.py
def update(self, bbox_xywh, confidences, ori_img):self.height, self.width = ori_img.shape[:2]# 提取每个bbox的featurefeatures = self._get_features(bbox_xywh, ori_img)# [cx,cy,w,h] -> [x1,y1,w,h]bbox_tlwh = self._xywh_to_tlwh(bbox_xywh)# 过滤掉置信度小于self.min_confidence的bbox,生成detectionsdetections = [Detection(bbox_tlwh[i], conf, features[i]) for i,conf in enumerate(confidences) if conf > self.min_confidence]# NMS (这里self.nms_max_overlap的值为1,即保留了所有的detections)boxes = np.array([d.tlwh for d in detections])scores = np.array([d.confidence for d in detections])indices = non_max_suppression(boxes, self.nms_max_overlap, scores)detections = [detections[i] for i in indices]...

2.3 卡尔曼滤波预测tracks

使用卡尔曼滤波预测前一帧中的tracks在当前帧的状态:

#  track.py
def predict(self, kf):"""Propagate the state distribution to the current time step using a Kalman filter prediction step.Parameters----------kf: The Kalman filter."""self.mean, self.covariance = kf.predict(self.mean, self.covariance)  # 预测self.age += 1  # 该track自出现以来的总帧数加1self.time_since_update += 1  # 该track自最近一次更新以来的总帧数加1

2.4 匈牙利算法关联匹配

首先对基于外观信息的欧式距离计算tracks和detections的代价矩阵,然后相继进行级联匹配和IOU匹配,最后得到当前帧的所有匹配对、未匹配的tracks以及未匹配的detections:

#  tracker.py
def _match(self, detections):def gated_metric(racks, dets, track_indices, detection_indices):"""基于外观信息和欧式距离,计算卡尔曼滤波预测的tracks和当前时刻检测到的detections的代价矩阵"""features = np.array([dets[i].feature for i in detection_indices])targets = np.array([tracks[i].track_id for i in track_indices]# 基于外观信息,计算tracks和detections的余弦距离代价矩阵cost_matrix = self.metric.distance(features, targets)# 基于欧式距离,过滤掉代价矩阵中一些不合适的项 (将其设置为一个较大的值)cost_matrix = linear_assignment.gate_cost_matrix(self.kf, cost_matrix, tracks, dets, track_indices, detection_indices)return cost_matrix# 区分开confirmed tracks和unconfirmed tracksconfirmed_tracks = [i for i, t in enumerate(self.tracks) if t.is_confirmed()]unconfirmed_tracks = [i for i, t in enumerate(self.tracks) if not t.is_confirmed()]# 对confirmd tracks进行级联匹配matches_a, unmatched_tracks_a, unmatched_detections = \linear_assignment.matching_cascade(gated_metric, self.metric.matching_threshold, self.max_age,self.tracks, detections, confirmed_tracks)# 对级联匹配中未匹配的tracks和unconfirmed tracks中time_since_update为1的tracks进行IOU匹配iou_track_candidates = unconfirmed_tracks + [k for k in unmatched_tracks_a ifself.tracks[k].time_since_update == 1]unmatched_tracks_a = [k for k in unmatched_tracks_a ifself.tracks[k].time_since_update != 1]matches_b, unmatched_tracks_b, unmatched_detections = \linear_assignment.min_cost_matching(iou_matching.iou_cost, self.max_iou_distance, self.tracks,detections, iou_track_candidates, unmatched_detections)# 整合所有的匹配对和未匹配的tracksmatches = matches_a + matches_bunmatched_tracks = list(set(unmatched_tracks_a + unmatched_tracks_b))return matches, unmatched_tracks, unmatched_detections# 级联匹配源码  linear_assignment.py
def matching_cascade(distance_metric, max_distance, cascade_depth, tracks, detections, track_indices=None, detection_indices=None):...unmatched_detections = detection_indicematches = []# 由小到大依次对每个level的tracks做匹配for level in range(cascade_depth):# 如果没有detections,退出循环if len(unmatched_detections) == 0:  break# 当前level的所有tracks索引track_indices_l = [k for k in track_indices if tracks[k].time_since_update == 1 + level]# 如果当前level没有track,继续if len(track_indices_l) == 0: continue# 匈牙利匹配matches_l, _, unmatched_detections = min_cost_matching(distance_metric, max_distance, tracks, detections, track_indices_l, unmatched_detections)matches += matches_lunmatched_tracks = list(set(track_indices) - set(k for k, _ in matches))return matches, unmatched_tracks, unmatched_detections

2.5 卡尔曼滤波更新

对于每个匹配成功的track,用其对应的detection进行更新,并处理未匹配tracks和detections:

#  tracker.py
def update(self, detections):"""Perform measurement update and track management.Parameters----------detections: List[deep_sort.detection.Detection]A list of detections at the current time step."""# 得到匹配对、未匹配的tracks、未匹配的dectectionsmatches, unmatched_tracks, unmatched_detections = self._match(detections)# 对于每个匹配成功的track,用其对应的detection进行更新for track_idx, detection_idx in matches:self.tracks[track_idx].update(self.kf, detections[detection_idx])# 对于未匹配的成功的track,将其标记为丢失for track_idx in unmatched_tracks:self.tracks[track_idx].mark_missed()# 对于未匹配成功的detection,初始化为新的trackfor detection_idx in unmatched_detections:self._initiate_track(detections[detection_idx])...

这篇关于匈牙利算法相关介绍的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/766730

相关文章

不懂推荐算法也能设计推荐系统

本文以商业化应用推荐为例,告诉我们不懂推荐算法的产品,也能从产品侧出发, 设计出一款不错的推荐系统。 相信很多新手产品,看到算法二字,多是懵圈的。 什么排序算法、最短路径等都是相对传统的算法(注:传统是指科班出身的产品都会接触过)。但对于推荐算法,多数产品对着网上搜到的资源,都会无从下手。特别当某些推荐算法 和 “AI”扯上关系后,更是加大了理解的难度。 但,不了解推荐算法,就无法做推荐系

性能测试介绍

性能测试是一种测试方法,旨在评估系统、应用程序或组件在现实场景中的性能表现和可靠性。它通常用于衡量系统在不同负载条件下的响应时间、吞吐量、资源利用率、稳定性和可扩展性等关键指标。 为什么要进行性能测试 通过性能测试,可以确定系统是否能够满足预期的性能要求,找出性能瓶颈和潜在的问题,并进行优化和调整。 发现性能瓶颈:性能测试可以帮助发现系统的性能瓶颈,即系统在高负载或高并发情况下可能出现的问题

水位雨量在线监测系统概述及应用介绍

在当今社会,随着科技的飞速发展,各种智能监测系统已成为保障公共安全、促进资源管理和环境保护的重要工具。其中,水位雨量在线监测系统作为自然灾害预警、水资源管理及水利工程运行的关键技术,其重要性不言而喻。 一、水位雨量在线监测系统的基本原理 水位雨量在线监测系统主要由数据采集单元、数据传输网络、数据处理中心及用户终端四大部分构成,形成了一个完整的闭环系统。 数据采集单元:这是系统的“眼睛”,

Hadoop数据压缩使用介绍

一、压缩原则 (1)运算密集型的Job,少用压缩 (2)IO密集型的Job,多用压缩 二、压缩算法比较 三、压缩位置选择 四、压缩参数配置 1)为了支持多种压缩/解压缩算法,Hadoop引入了编码/解码器 2)要在Hadoop中启用压缩,可以配置如下参数

康拓展开(hash算法中会用到)

康拓展开是一个全排列到一个自然数的双射(也就是某个全排列与某个自然数一一对应) 公式: X=a[n]*(n-1)!+a[n-1]*(n-2)!+...+a[i]*(i-1)!+...+a[1]*0! 其中,a[i]为整数,并且0<=a[i]<i,1<=i<=n。(a[i]在不同应用中的含义不同); 典型应用: 计算当前排列在所有由小到大全排列中的顺序,也就是说求当前排列是第

csu 1446 Problem J Modified LCS (扩展欧几里得算法的简单应用)

这是一道扩展欧几里得算法的简单应用题,这题是在湖南多校训练赛中队友ac的一道题,在比赛之后请教了队友,然后自己把它a掉 这也是自己独自做扩展欧几里得算法的题目 题意:把题意转变下就变成了:求d1*x - d2*y = f2 - f1的解,很明显用exgcd来解 下面介绍一下exgcd的一些知识点:求ax + by = c的解 一、首先求ax + by = gcd(a,b)的解 这个

综合安防管理平台LntonAIServer视频监控汇聚抖动检测算法优势

LntonAIServer视频质量诊断功能中的抖动检测是一个专门针对视频稳定性进行分析的功能。抖动通常是指视频帧之间的不必要运动,这种运动可能是由于摄像机的移动、传输中的错误或编解码问题导致的。抖动检测对于确保视频内容的平滑性和观看体验至关重要。 优势 1. 提高图像质量 - 清晰度提升:减少抖动,提高图像的清晰度和细节表现力,使得监控画面更加真实可信。 - 细节增强:在低光条件下,抖

sqlite3 相关知识

WAL 模式 VS 回滚模式 特性WAL 模式回滚模式(Rollback Journal)定义使用写前日志来记录变更。使用回滚日志来记录事务的所有修改。特点更高的并发性和性能;支持多读者和单写者。支持安全的事务回滚,但并发性较低。性能写入性能更好,尤其是读多写少的场景。写操作会造成较大的性能开销,尤其是在事务开始时。写入流程数据首先写入 WAL 文件,然后才从 WAL 刷新到主数据库。数据在开始

【数据结构】——原来排序算法搞懂这些就行,轻松拿捏

前言:快速排序的实现最重要的是找基准值,下面让我们来了解如何实现找基准值 基准值的注释:在快排的过程中,每一次我们要取一个元素作为枢纽值,以这个数字来将序列划分为两部分。 在此我们采用三数取中法,也就是取左端、中间、右端三个数,然后进行排序,将中间数作为枢纽值。 快速排序实现主框架: //快速排序 void QuickSort(int* arr, int left, int rig

poj 3974 and hdu 3068 最长回文串的O(n)解法(Manacher算法)

求一段字符串中的最长回文串。 因为数据量比较大,用原来的O(n^2)会爆。 小白上的O(n^2)解法代码:TLE啦~ #include<stdio.h>#include<string.h>const int Maxn = 1000000;char s[Maxn];int main(){char e[] = {"END"};while(scanf("%s", s) != EO