supervision – 一个可重复使用的计算机视觉工具开源项目

Supervision 提供直观易用的工具,让您快速构建计算机视觉应用,无需深入了解底层技术细节。计算机视觉加速器,助力项目快速落地。节省大量开发时间,专注于核心算法的创新。

特性

Supervision 简化了使用视觉模型的过程。提供了与常见模型库的连接器、多种可视化工具(标注器)、强大的后处理功能,并且学习曲线较为平缓。

  1. 加载图像或视频
  2. 加载模型
  3. 运行模型
  4. 标注

核心概念

  • 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

官方链接

Comments

No comments yet. Why don’t you start the discussion?

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注