Save Detections

保存detections为csv文件和json文件。

import cv2 as cv
import supervision as sv
from ultralytics import YOLO

model_path=r'F:\python\yolov8\yolov8n.pt'
model=YOLO(model_path)

video_path='vehicles.mp4'
# 获取视频帧生成器
frame_generator=sv.get_video_frames_generator(video_path)
# 保存为csv文件
with sv.CSVSink('s_detections.csv') as sink:
# 保存为json文件
# with sv.JSONSink('s_detections.json') as sink:
    for frame_index,frame in enumerate(frame_generator):
        results=model(frame)[0]
        detections=sv.Detections.from_ultralytics(results)
        # {"frame_index":frame_index} 这是自定义的数据
        sink.append(detections,{"frame_index":frame_index})