lightly-ai/lightly-studio

Curate, Annotate, and Manage Your Data in LightlyStudio.

What it solves

LightlyStudio 是一个开源工具,统一了为计算机视觉模型准备数据的碎片化工作流。它解决了管理、整理和标注大型图像与视频数据集的难题,同时在单一界面中评估模型预测与真实标注的差异。

How it works

该工具作为基于浏览器的应用运行在用户本机上。用户通过 Python 接口将数据(本地路径或 S3、GCS 等云存储)加载到本地 DuckDB 数据库。服务器启动后,用户可以通过 GUI 探索、过滤和管理数据。它利用 Rust 提供高性能,能够在消费级硬件上处理如 COCO、ImageNet 等大型数据集。

Who it’s for

它面向从事计算机视觉的机器学习工程师和数据科学家,尤其是需要整理高质量训练集、管理目标检测与分割标注、以及分析模型失效模式的人员。

Highlights

  • Unified Workflow: 将整理、标注和模型评估合并在一个工具中。
  • Multi-modal Support: 支持图像和视频数据集。
  • Advanced Sampling: 包含选择具代表性和多样化样本的策略,以降低标注成本。
  • Flexible Data Loading: 支持本地文件和主要云存储提供商(AWS S3、GCS、Azure)。
  • Extensible: 支持如 SAM 自动标注等插件。
  • Broad Format Support: 兼容 COCO 和 YOLO 格式。