如何在YOLOv8中实现已识别对象5秒内不重复输出ID与类别？

核心思路是用字典存储已输出对象的ID及对应的输出时间戳，针对每个跟踪ID单独控制输出冷却，避免同一对象短时间内重复输出。

实现代码 #

import cv2
from ultralytics import YOLO
import time

# 加载YOLOv8模型（支持跟踪模式）
model = YOLO('yolov8n.pt')

# 存储已输出对象的ID与上次输出时间：{对象ID: 上次输出时间戳}
output_history = {}
# 输出冷却时间（单位：秒）
COOLDOWN_DURATION = 5

# 读取视频源（本地路径/摄像头ID）
video_cap = cv2.VideoCapture('your_video_file.mp4')

while video_cap.isOpened():
    ret, frame = video_cap.read()
    if not ret:
        break

    # 运行YOLOv8跟踪，persist=True保证ID在帧间稳定
    track_results = model.track(frame, persist=True, verbose=False)

    # 仅当存在跟踪ID时处理
    if track_results[0].boxes.id is not None:
        # 提取当前帧的对象ID、类别索引、类别名称
        obj_ids = track_results[0].boxes.id.int().cpu().tolist()
        class_indices = track_results[0].boxes.cls.int().cpu().tolist()
        class_names = track_results[0].names

        current_timestamp = time.time()

        for obj_id, cls_idx in zip(obj_ids, class_indices):
            cls_name = class_names[cls_idx]
            # 判断是否满足输出条件：未记录过 或 已超过冷却时间
            if obj_id not in output_history or (current_timestamp - output_history[obj_id]) > COOLDOWN_DURATION:
                print(f"对象信息: ID={obj_id}, 类别={cls_name}")
                # 更新该ID的输出时间
                output_history[obj_id] = current_timestamp

    # 可选：显示标注后的帧
    annotated_frame = track_results[0].plot()
    cv2.imshow('YOLOv8 Tracking', annotated_frame)

    # 按Q键退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
video_cap.release()
cv2.destroyAllWindows()

关键逻辑说明 #

1. 字典存储优势：相比全局数组，字典能为每个跟踪ID单独记录输出时间，精准控制单对象的冷却周期，不会出现全局清空导致的误判。
2. 稳定跟踪ID：model.track()中persist=True参数确保对象在移动、短暂遮挡后仍保持原有ID，避免ID频繁变化导致的重复输出。
3. 时间戳判断：用time.time()获取秒级时间戳，通过时间差判断是否超过5秒冷却时间，逻辑简单可靠。

为何之前的方案不适用 #

• 销毁窗口：与输出控制逻辑完全无关，无法解决重复输出问题。
• 计数类名：仅能按类别限制输出，无法区分同一类别下的不同对象，会导致同一类别多个对象被错误限制。
• 数组存ID后全局清空：全局清空会重置所有ID的冷却状态，无法保证单个对象的5秒冷却要求。

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