产品推荐 2026-07-27

cosmos-curator:AI视频数据集生成工具(源码)

AI驱动的视频处理系统,用于分析、整理大规模视频数据集。

开源软件跨平台工具

产品简介

cosmos-curator NVIDIA Cosmos Curator是一个利用先进AI模型和分布式计算的视频处理系统。它能高效完成视频分割、标注、去重和数据集生成,专为需要大规模生成高质量训练数据的研究人员和团队设计。

cosmos-curator 主界面

核心功能

  • 视频处理:高效进行视频分割、标注、过滤、去重和数据集生成
  • AI分析:使用多种模型进行先进的视频内容分析
  • 分布式计算:基于Cosmos-Xenna和Ray实现可扩展的分布式处理
  • 云平台集成:支持多种部署平台
  • 模块化流水线:架构灵活可扩展

快速开始

  1. 同步并初始化Git子模块:运行 `git submodule update --init --recursive`。
  2. 查阅文档:参考 `docs/client/end-user-guide.md` 进行环境配置。
  3. 选择运行方式:使用本地CLI、Slurm集群或NVIDIA云函数等工具启动处理流水线。

界面预览

与其他方案对比

对比项cosmos-curator同类方案
处理规模专为大规模视频数据集设计,利用分布式计算实现高效处理。传统工具通常针对单机或少量视频,难以扩展至海量数据。
AI模型集成深度集成多种先进AI模型,支持分析、理解、标注等高级处理。多数通用视频工具仅提供基础编辑功能,缺乏智能分析能力。
数据生成生成的“数据集”概念较弱,更侧重于完成单次处理任务。
分布式支持可能不支持或仅支持简单的并行任务,缺乏成熟的分布式框架。

适合谁用

  • 需要为AI模型生成大规模高质量训练数据的研究人员。
  • 从事计算机视觉或视频分析项目,需要构建自动化处理流水线的工程师。
  • 在具备GPU集群的机构中,负责视频数据预处理和管理的团队。

下载与版本

当前最新版本为 v2.2.0,发布于 2026-06-11。支持 Windows、macOS、Linux。

⚠️ 注意:本项目仅提供源代码下载,不包含可直接运行的安装包(exe/dmg/apk 等。下载后需要自行编译构建才能使用,适合有一定开发经验的用户。如果你不熟悉编译流程,建议寻找同类开箱即用的替代方案。

注意事项

  • 本工具仅供学习和研究使用,请遵守相关法律法规。
  • 请尊重内容创作者版权,勿将下载内容用于商业或非法传播。
  • 项目源码地址:https://github.com/NVIDIA/cosmos-curator