目标跟踪中的匈牙利算法

🕗 发布于 2024-09-21 09:45 目标跟踪算法 人工智能

从数学角度来看，线性分配问题（也称为匈牙利算法或指派问题）是一个经典的优化问题，其目的是在两个集合之间找到最佳匹配，使得总成本最小。我们可以将其形式化为一个二分图的最小权匹配问题。

数学背景

假设我们有两个集合 ( A ) 和 ( B )，其中 ( A ) 有 ( m ) 个元素，( B ) 有 ( n ) 个元素。我们有一个 ( m \times n ) 的成本矩阵 ( C )，其中 ( C[i, j] ) 表示将 ( A ) 中的第 ( i ) 个元素分配给 ( B ) 中的第 ( j ) 个元素的成本。

目标是找到一个匹配 $\subseteq A \times B )$ ，使得总成本

$\sum_{(i, j) \in M} C[i, j]$

最小，并且每个元素最多匹配一次。

算法解释

1. 空成本矩阵的处理

如果成本矩阵为空（即没有元素），则直接返回空匹配和未匹配的索引。

if cost_matrix.size == 0:
    return np.empty((0, 2), dtype=int), tuple(range(cost_matrix.shape[0])), tuple(range(cost_matrix.shape[1]))

2. 使用 LAPJV 算法

如果选择使用 LAPJV 算法（来自 lap 库），则执行以下步骤：

_, x, y = lap.lapjv(cost_matrix, extend_cost=True, cost_limit=thresh)

lap.lapjv 函数返回三个数组：
- 第一个数组（未使用）是总成本。
- x 数组表示每个 ( A ) 中元素的匹配结果，如果未匹配则为 -1。
- y 数组表示每个 ( B ) 中元素的匹配结果，如果未匹配则为 -1。

matches = [[ix, mx] for ix, mx in enumerate(x) if mx >= 0]
unmatched_a = np.where(x < 0)[0]
unmatched_b = np.where(y < 0)[0]

matches 是一个列表，包含所有有效的匹配对。
unmatched_a 和 unmatched_b 分别是未匹配的 ( A ) 和 ( B ) 中的索引。

3. 使用 SciPy 的线性和分配算法

如果选择使用 SciPy 的 linear_sum_assignment 算法，则执行以下步骤：

x, y = scipy.optimize.linear_sum_assignment(cost_matrix)

scipy.optimize.linear_sum_assignment 返回两个数组：
- x 表示 ( A ) 中元素的匹配索引。
- y 表示 ( B ) 中元素的匹配索引。

matches = np.asarray([[x[i], y[i]] for i in range(len(x)) if cost_matrix[x[i], y[i]] <= thresh])

这行代码筛选出所有匹配成本不超过阈值的匹配对。

if len(matches) == 0:
    unmatched_a = list(np.arange(cost_matrix.shape[0]))
    unmatched_b = list(np.arange(cost_matrix.shape[1]))
else:
    unmatched_a = list(set(np.arange(cost_matrix.shape[0])) - set(matches[:, 0]))
    unmatched_b = list(set(np.arange(cost_matrix.shape[1])) - set(matches[:, 1]))

如果没有有效匹配，所有元素都未匹配。
否则，计算未匹配的 ( A ) 和 ( B ) 中的索引。

线性分配问题通过最小化总成本来找到两个集合之间的最佳匹配。上述函数实现了这个过程，并提供了两种不同的算法选择（LAPJV 和 SciPy 的线性和分配算法）。通过筛选出有效的匹配对和未匹配的元素索引，函数可以有效地处理目标跟踪等应用中的匹配问题。

匹配过程的计算涉及到解决一个优化问题，即找到两个集合之间的最佳匹配，使得总成本最小。这个问题可以用匈牙利算法（Hungarian Algorithm）或其他线性分配算法来解决。下面详细解释这些算法是如何计算匹配的。

1. 匈牙利算法（Hungarian Algorithm）

匈牙利算法是一个经典的解决线性分配问题的算法，具有多项式时间复杂度。它的基本思想是通过逐步修改成本矩阵，找到一个零成本的完美匹配。具体步骤如下：

1.1 构造初始零矩阵

从每一行中减去该行的最小值，使得每一行至少有一个零。
从每一列中减去该列的最小值，使得每一列至少有一个零。

1.2 覆盖零

尽可能少地画横线和竖线覆盖所有的零。
如果所需的线条数量等于行（或列）的数量，则找到最优匹配。

1.3 调整矩阵

如果线条数量少于行（或列）的数量，找出未被覆盖的最小元素，将其从所有未覆盖元素中减去，并将其加到所有被两条线覆盖的元素上，重复步骤1.2。

2. LAPJV 算法

LAPJV（Jonker-Volgenant Algorithm）是匈牙利算法的一个高效变种，适用于稀疏矩阵。它通过扩展成本矩阵和使用增广路径技术来找到最优匹配。

2.1 初始化

初始化标号（labels）和匹配。
标号表示当前的成本调整，匹配表示当前的部分匹配。

2.2 构造增广路径

从未匹配的行开始，寻找增广路径。
增广路径是从未匹配的行到未匹配的列的一条路径，通过调整标号来找到更低成本的匹配。

2.3 更新匹配

使用增广路径更新匹配。
重复构造增广路径和更新匹配，直到找到最优匹配。

3. SciPy 的线性和分配算法

SciPy 的 linear_sum_assignment 函数实现了匈牙利算法。它通过以下步骤来计算匹配：

3.1 构造初始零矩阵

从每一行中减去该行的最小值。
从每一列中减去该列的最小值。

3.2 覆盖零

使用最少的线条覆盖所有的零。

3.3 调整矩阵

找到未覆盖的最小元素，调整矩阵。
重复覆盖零和调整矩阵，直到找到最优匹配。

示例

假设我们有如下成本矩阵：

$\begin{bmatrix} 4 & 1 & 3 \\ 2 & 0 & 5 \\ 3 & 2 & 2 \end{bmatrix}$

使用匈牙利算法计算匹配

构造初始零矩阵：
- 行减去最小值： $\begin{bmatrix} 3 & 0 & 2 \\ 2 & 0 & 5 \\ 1 & 0 & 0 \end{bmatrix}$
- 列减去最小值： $\begin{bmatrix} 3 & 0 & 2 \\ 1 & 0 & 5 \\ 0 & 0 & 0 \end{bmatrix}$
覆盖零：
- 使用最少的线条覆盖所有的零。
调整矩阵：
- 找到未覆盖的最小元素，调整矩阵。
找到匹配：
- 通过增广路径找到最优匹配。

最终匹配结果可能是 ((0, 1), (1, 0), (2, 2))，总成本为 (1 + 2 + 2 = 5)。

匹配过程的核心是通过调整成本矩阵，使得在零成本上找到一个完美匹配。匈牙利算法和 LAPJV 算法都通过不同的路径和技术来实现这一目标。SciPy 的 linear_sum_assignment
函数实现了匈牙利算法，提供了一个高效的解决方案。

在目标跟踪问题中，目标框之间的匹配通常是通过计算每个目标框之间的相似度或距离来实现的。以下是一个常见的匹配过程，结合了匈牙利算法或其他线性分配算法来找到最佳匹配。

1. 定义成本矩阵

首先，我们需要定义一个成本矩阵 $C$ ，其中每个元素 $C [i, j]$ 表示第 $i$ 个目标框和第 $j$ 个目标框之间的匹配成本。匹配成本可以基于多种度量方式，例如欧氏距离、IoU（Intersection over Union）、马氏距离等。

1.1 欧氏距离

如果目标框的位置用中心点表示，可以计算中心点之间的欧氏距离：

$\sqrt{(x_i - x_j)^2 + (y_i - y_j)^2}$

1.2 IoU（Intersection over Union）

IoU 是目标跟踪中常用的度量方式，表示两个边界框的交集与并集之比：

$\text{IoU}(B_i, B_j)$

其中， $B_i$ 和 $B_j$ 分别是第 $i$ 个和第 $j$ 个目标框。

2. 构造成本矩阵

假设我们有两个集合：上一帧的目标框集合 $A$ 和当前帧的目标框集合 $B$ 。我们构造一个成本矩阵 $C$ ，其中 $C [i, j]$ 表示 $A$ 中的第 $i$ 个目标框和 $B$ 中的第 $j$ 个目标框之间的匹配成本。

3. 计算匹配

使用匈牙利算法或其他线性分配算法来计算最佳匹配。下面是一个使用 scipy.optimize.linear_sum_assignment 函数的示例：

import numpy as np
from scipy.optimize import linear_sum_assignment

# 假设 cost_matrix 是预先计算好的成本矩阵
cost_matrix = np.array([
    [4, 1, 3],
    [2, 0, 5],
    [3, 2, 2]
])

# 使用匈牙利算法计算最佳匹配
row_ind, col_ind = linear_sum_assignment(cost_matrix)

# row_ind 和 col_ind 分别表示最佳匹配的行索引和列索引
matches = list(zip(row_ind, col_ind))
print("匹配结果:", matches)

4. 处理未匹配的目标框

在实际应用中，可能会有一些目标框没有找到匹配。我们需要处理这些未匹配的目标框：

# 找到未匹配的目标框
unmatched_a = list(set(range(cost_matrix.shape[0])) - set(row_ind))
unmatched_b = list(set(range(cost_matrix.shape[1])) - set(col_ind))

print("未匹配的A中的目标框索引:", unmatched_a)
print("未匹配的B中的目标框索引:", unmatched_b)

5. 更新目标状态

根据匹配结果更新目标状态。例如，可以更新目标的轨迹、位置等信息。

示例总结

假设我们有以下成本矩阵：

$\begin{bmatrix} 4 & 1 & 3 \\ 2 & 0 & 5 \\ 3 & 2 & 2 \end{bmatrix}$

通过 linear_sum_assignment 函数，我们可以计算出最佳匹配：

row_ind, col_ind = linear_sum_assignment(cost_matrix)
matches = list(zip(row_ind, col_ind))

匹配结果可能是 ((0, 1), (1, 0), (2, 2))。

通过定义成本矩阵并使用匈牙利算法或其他线性分配算法，我们可以在目标跟踪中找到最佳的目标框匹配。匹配过程包括计算成本矩阵、使用优化算法计算最佳匹配、处理未匹配的目标框以及更新目标状态。这些步骤可以帮助我们在视频帧之间有效地跟踪目标。

目标检测中的非极大值抑制（Non-Maximum Suppression, NMS）和匈牙利算法解决的匹配问题是两个不同的概念，尽管它们都涉及到目标框的处理，但它们的应用场景和目的不同。

非极大值抑制（NMS）

NMS 是一种后处理技术，用于在目标检测中消除多余的检测框。目标检测算法通常会在同一目标上产生多个重叠的检测框，NMS 通过以下步骤来保留最好的检测框并去除冗余的框：

排序：根据检测框的置信度得分对所有检测框进行排序，从高到低。
选择最高得分框：选择得分最高的检测框，并将其加入最终的检测结果中。
移除重叠框：计算其他检测框与当前选择框的 IoU（Intersection over Union），如果 IoU 超过预设的阈值，则将这些重叠的检测框移除。
重复：重复步骤 2 和 3，直到没有剩余的检测框。

NMS 的目的是减少冗余检测框，保留最有可能的目标框。

匈牙利算法

匈牙利算法用于解决线性分配问题，目的是在两个集合之间找到最佳匹配，使得总的匹配成本最小。它通常用于目标跟踪中的目标框匹配，具体步骤如前面所述，包括构造成本矩阵、使用匈牙利算法计算最佳匹配以及处理未匹配的目标框。

NMS 和匈牙利算法的区别

目的：
- NMS：用于消除冗余检测框，保留最有可能的目标框。
- 匈牙利算法：用于在两个集合之间找到最佳匹配，使得总的匹配成本最小。
应用场景：
- NMS：主要用于目标检测的后处理阶段。
- 匈牙利算法：主要用于目标跟踪中的目标框匹配。
处理方式：
- NMS：基于检测框的置信度得分和 IoU 进行排序和移除。
- 匈牙利算法：基于成本矩阵和优化算法找到最佳匹配。

示例

假设我们在目标检测中得到了以下检测框和对应的置信度得分：

框 1：得分 0.9
框 2：得分 0.8
框 3：得分 0.7

这些检测框有一定的重叠度。NMS 的工作流程如下：

排序：框 1、框 2、框 3。
选择最高得分框：选择框 1。
移除重叠框：计算框 1 与框 2、框 3 的 IoU，如果 IoU 超过阈值，将这些框移除。
重复：选择下一个最高得分框（框 2），重复步骤 3，直到没有剩余的检测框。

最终，NMS 将保留最有可能的目标框，减少冗余。

非极大值抑制（NMS）和匈牙利算法解决的匹配问题是不同的技术，尽管它们都涉及到目标框的处理。NMS 主要用于目标检测的后处理阶段，以消除冗余检测框，而匈牙利算法主要用于目标跟踪中的目标框匹配，以找到最佳匹配。两者在目的、应用场景和处理方式上都有显著区别。

原文地址：https://blog.csdn.net/weixin_67051070/article/details/142387742

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：计算数学精解【15】- flux机器学习精解【1】
下一篇：解决 iOS App Tracking Transparency 权限问题

JACM23 - A New Algorithm for Euclidean Shortest Paths in the Plane
本文关注的问题为计算几何学中的经典问题，即「在平面上给定一组两两不相交的多边形障碍物，寻找两点之间避开所有障碍物的欧几里得最短路径」，简单理解就是「含多边形障碍物的两点最短路问题」。
阅读更多2024-09-22
linux如何对c++进行内存分析
linux如何对c++进行内存分析
阅读更多2024-09-22
Mina protocol - 体验教程
零知识证明是一种密码学协议，允许证明者在不泄露任何额外信息的情况下，向验证者证明自己知道某个特定的秘密或信息。：zkCircuits 处理初始状态（State0），接受公共和私有输入，生成交易证明（T
阅读更多2024-09-22
springMvc的初始配置
mapper层（另外创建DataSourceJavaConfig ）3.统一放入IOC容器中。controller层。
阅读更多2024-09-22
17【Protues单片机仿真】基于51单片机的太阳能智能谷物翻晒机器人
基于51单片机，避障，低于50CM报警，LED灯亮起，自动翻晒用光敏电阻，光照强度大，电机转动，相当于翻晒粮食，远程控制用按键代替，按下去电机就转动，相当于翻晒粮食，然后光照强度和超声波的距离都在LC
阅读更多2024-09-22
大数据新视界 --大数据大厂之 Node.js 与大数据交互：实现高效数据处理
本文深入阐释 Node.js 与大数据交互。介绍 Node.js 优势，包括高效事件驱动和非阻塞 I/O 及丰富模块生态。阐述与大数据存储系统连接、与处理框架集成方式，通过实时数据分析平台和可视化应用
阅读更多2024-09-22
【C++】—— string模拟实现
学习了string的使用，总感觉了解不是很深厚；自己模拟实现string类来帮助自己理解。
阅读更多2024-09-22
iPhone16，超先进摄像头系统？丝滑的相机控制
iPhone 16将于9月20号正式开售，这篇文章我们来看下iPhone 16 在影像方面，有哪些升级和新feature。芯片：采用第二代 3纳米芯片，A18。摄像头配置：iPhone 16前置：索尼
阅读更多2024-09-22
lora 微调3B模型微调前有5G 量化f16 后最后导出模型容量变小了只有2G了，为什么？
通过量化、LoRA微调、剪枝和存储格式优化等方法的组合，最终导出的模型容量通常会比原始模型显著减少。从5G减少到2G是这些多重因素共同作用的结果。如果你有特定的工具或框架，可以查看其文档了解详细的模型
阅读更多2024-09-22
互联网技术的持续演进：从现在到未来
互联网技术的发展在过去的十年里已经发生了翻天覆地的变化，未来的进展只会更加令人期待。从5G、人工智能到物联网、区块链，各种新兴技术将会继续推动社会的数字化转型。无论是个人、企业还是政府，都将从这些技术
阅读更多2024-09-22