Centroid-Tracking质心跟踪排序匹配逻辑与相关代码作用咨询

距离矩阵D的基础含义 #

首先dist.cdist生成的距离矩阵D维度为M×N：

• M为当前已跟踪对象的总数量，每一行对应一个已存历史质心
• N为当前帧新检测到的质心总数量，每一列对应一个新检测质心
• 元素D[i][j]表示第i个已存质心和第j个新质心的欧氏距离

两行匹配代码的具体作用 #

1. rows = D.min(axis=1).argsort() #

• D.min(axis=1)会按行计算最小值，输出长度为M的数组，每个元素对应某个已存质心和所有新质心的最小匹配距离
• argsort()将上述最小距离从小到大排序，返回对应的行索引数组。排在数组越靠前的索引，对应已存质心的最小匹配距离越小，匹配置信度越高。
• 核心作用是确定匹配优先级，优先处理匹配度最高的配对，避免低置信度匹配抢占高置信度的配对资源。

2. cols = D.argmin(axis=1)[rows] #

• D.argmin(axis=1)会按行计算最小值对应的列索引，输出长度为M的数组，每个元素对应某个已存质心的最佳候选新质心的索引
• 用前一步得到的优先级索引rows对上述数组重排，得到和rows顺序一一对应的候选新质心索引数组。
• 核心作用是将每个已存质心的最佳候选匹配和优先级顺序做对应，为后续循环匹配做准备。

该设计的必要性 #

如果不做排序直接逐行取最小距离匹配，会出现多个已存质心的最佳候选都是同一个新质心的情况，导致匹配冲突。这两行代码配合后续的usedRows、usedCols去重逻辑，本质是贪心实现的二分图最优匹配，保证每个新质心只会分配给和它距离最近的已存质心，最大程度降低总匹配误差，避免匹配冲突。

内容的提问来源于stack exchange，提问作者Mohammed B.