YOLOv8 track功能对象ID自动变更问题求助

问题描述 #

使用YOLOv8的track功能时，可正常获取目标的边界框和类别信息，但存在以下问题：当某一目标未被检测到时，其余目标的ID会自动递补变更。例如当前追踪ID为1、2、3、4，若目标2丢失，原本的3、4会被改为2、3，最终ID变为1、2、3，导致需手动判断哪个目标消失，操作繁琐。

注意到开启show=True时，可视化窗口中即使目标丢失或离开画面，仍能显示正确的ID，但查阅官方文档未找到对应的API调用方法。另外track返回的结果对象包含sort和index方法，不确定如何利用。

示例代码：

from ultralytics import YOLO
model = YOLO("yolov8n-seg.pt")
results = model.track(img, conf=0.20, show=True, persist=True, tracker="botsort.yaml", verbose=False )
print(results[0].boxes.data[:,4])

输出情况：

# 目标全部存在时：
1,2,3,4
# 目标2未被检测到时：
1,2,3

解决方法 #

1. 调整追踪器配置文件参数 #

你使用的是botsort.yaml追踪器，打开该配置文件，修改track_buffer参数。这个参数控制目标丢失后保留其ID的帧数，默认值可能过小，建议根据你的视频帧率调大（比如设为30），这样目标短暂丢失时ID不会立刻被回收复用，避免ID被重新分配。

2. 使用boxes.id属性获取原始追踪ID #

不要直接从results[0].boxes.data[:,4]提取ID，改用results[0].boxes.id属性。该属性返回的是未被重新分配的原始追踪ID张量，即使目标丢失后重新出现，ID也会保持一致。示例代码：

if results[0].boxes.id is not None:
    # 转换为CPU上的numpy数组便于处理
    track_ids = results[0].boxes.id.cpu().numpy()
    print("当前追踪ID：", track_ids)

3. 手动维护全局ID映射表 #

如果上述方法仍无法满足需求，可自行维护一个全局集合记录所有出现过的追踪ID，每次追踪时对比当前检测到的ID，自动找出消失的目标ID：

# 全局变量，记录所有曾出现过的活跃ID
active_ids = set()

results = model.track(img, conf=0.20, persist=True, tracker="botsort.yaml", verbose=False )
if results[0].boxes.id is not None:
    current_ids = set(results[0].boxes.id.cpu().numpy())
    # 计算消失的ID：之前存在但当前未检测到的
    lost_ids = active_ids - current_ids
    if lost_ids:
        print("消失的目标ID：", lost_ids)
    # 更新活跃ID集合，加入新出现的ID
    active_ids.update(current_ids)

4. 关于sort和index方法的说明 #

• sort方法：可对检测框按指定字段排序，比如按ID排序：results[0].boxes.sort("id")，排序后便于按ID顺序处理目标。
• index方法：主要用于批量处理多帧结果时，通过索引获取对应帧的结果，比如results.index(0)获取第一帧的追踪结果，日常单帧追踪场景中使用较少。

内容的提问来源于stack exchange，提问作者MLRAL