你是否曾为训练完一个目标检测模型后,却卡在画框、算指标、导视频、做可视化上而头疼?Supervision 就是为此而生——它不是模型,也不是框架,而是一套开箱即用、高度模块化、文档极友好的 Python 工具库,帮你把“模型输出”快速变成“可交付成果”。无论你是刚跑通 YOLO 的学生,还是要上线工业质检系统的工程师,Supervision 都能省下你 70% 的重复编码时间。
核心功能
- 智能标注渲染:一行代码即可在图像/视频上绘制检测框、分割掩码、旋转框(OBB)、关键点,支持自定义颜色、字体、透明度和标签格式,告别手写 OpenCV 绘图逻辑
- 开箱即用的评估体系:内置 COCO、Pascal VOC 等主流协议的 mAP、Recall、Precision 计算器,支持按类别、置信度阈值、IoU 范围灵活分析,结果可直接导出为 Pandas DataFrame 或 HTML 报告
- 轻量级跟踪增强:无缝集成 ByteTrack、BoT-SORT 等主流追踪器,自动关联跨帧 ID,支持轨迹绘制、速度估算、出入区域统计,无需从零搭建 tracker pipeline
- 低代码视频处理流水线:通过
VideoSink和FrameAnnotator构建清晰的数据流,轻松实现“读帧→推理→标注→写回”的端到端视频处理,支持 RTSP、MP4、摄像头多源输入 - 数据格式自由转换:一键导入/导出 COCO JSON、YOLO TXT、Pascal VOC XML、Roboflow 格式等 8+ 种标注标准,团队协作或模型迁移时再不用手动写转换脚本
- 可扩展的处理模块:提供
Detections统一数据结构封装所有检测结果,支持链式过滤(如.with_nms()、.filter_by_confidence(0.5))、坐标变换、裁剪、缩放等操作,逻辑清晰、调试直观
适合哪些人用
如果你是以下角色之一,Supervision 几乎是必试工具:计算机视觉初学者(快速验证想法、避免陷入底层绘图细节);算法研究员(专注模型改进,把评估和可视化交给成熟工具);AI 应用开发者(需要快速交付带标注界面的 Demo 或内部系统);技术产品经理/低代码平台构建者(基于其模块组装可视化组件或自动化报告模块)。它不替代 PyTorch/TensorFlow,而是站在巨人肩膀上,让你更高效地“交付价值”。
快速上手
安装只需一条命令:
pip install supervision
5 行代码即可完成目标检测结果可视化:
from supervision import Detections, BoxAnnotator
from PIL import Image
# 假设你已有模型输出:boxes, scores, class_ids
detections = Detections(xyxy=boxes, confidence=scores, class_id=class_ids)
annotator = BoxAnnotator()
annotated_image = annotator.annotate(scene=Image.open("input.jpg"), detections=detections)
官方提供超 30 个 Jupyter Notebook 教程(含 YOLOv8、RT-DETR、Segment Anything 集成示例),中文社区已整理翻译版,GitHub Wiki 和 Discord 社区响应活跃。
项目信息
roboflow/supervision
GitHub
We write your reusable computer vision tools. 💜
38.7k
今日 +59 stars today
Stars
3.5k
Forks
Python
MIT
编程语言:Python|GitHub Star 数:38,744|开源协议:MIT|GitHub 项目地址
它不是另一个“大而全”的框架,而是一把精准、锋利、随时可用的瑞士军刀——当你厌倦了反复造轮子,Supervision 就是你今天值得安装的第一个 CV 工具库。


