Supervision 提供直观易用的工具,让您快速构建计算机视觉应用,无需深入了解底层技术细节。计算机视觉加速器,助力项目快速落地。节省大量开发时间,专注于核心算法的创新。
特性
Supervision 简化了使用视觉模型的过程。提供了与常见模型库的连接器、多种可视化工具(标注器)、强大的后处理功能,并且学习曲线较为平缓。
- 加载图像或视频
- 加载模型
- 运行模型
- 标注
核心概念
- sv.Detections: 这是一个通用类,适用于对象检测和分割模型,包含字段:xyxy(坐标)、mask(掩码)、class_id(类别ID)、tracker_id(追踪ID)、data(数据)。
- import supervision as sv: 全局范围内可以使用所有有用的函数。
- 模型选择: 加载流行、经过微调的模型或 Universe 模型。
- sv.Detections.from_X: 从11个来源之一加载数据。
- 标注器: 使用20种标注器之一绘制检测结果。
安装
支持 Python 3.8 及以上版本
# 无图形用户界面 (GUI),更加轻量级,适用于服务器端应用程序。
pip install supervision
# 此版本包括 OpenCV 的 GUI 组件,允许您在屏幕上显示图像和视频。
pip install "supervision[desktop]"
$ conda
conda install -c conda-forge supervision
$ mamba
mamba install -c conda-forge supervision