NVIDIA/cosmos-curator

NVIDIA/cosmos-curator

Description: Cosmos Curator is a powerful video curation system that processes, analyzes, and organizes video content using advanced AI models and distributed computing.

Language: Python

License: Apache-2.0

Stars: 204

Forks: 30

Open issues: 0

Created: 2025-05-28T00:44:30Z

Pushed: 2026-06-05T13:51:53Z

Default branch: main

Fork: no

Archived: no

README:

Product Website

Cosmos Curator

A powerful video curation system that processes, analyzes, and organizes video content using advanced AI models and distributed computing.

Important

Please run git submodule sync if you have cloned the repository before and just pulled the latest update. We updated the URL for cosmos-xenna submodule on 08/04/2025.

Overview

Cosmos Curator is a comprehensive solution for video processing and curation using state-of-the-art AI models, which powers the training data generation for Cosmos at NVIDIA. It is built on top of a framework optimized for GPU-accelerated streaming pipeline, which is now open-sourced independently as Cosmos-Xenna.

Features

• Video Processing: Efficient video splitting, annotation, filtering, deduplication, and dataset generation
• AI-Powered Analysis: Advanced video analysis using multiple model families
• Distributed Computing: Scalable processing using Cosmos-Xenna built on top of Ray
• Cloud Integration: Support for various platforms
• Pipeline System: Modular and extensible pipeline architecture

Documentation

Comprehensive documentation is available under [docs/](docs/README.md) directory.

User Documentation

• [End User Guide](docs/client/end-user-guide.md) - instructions to setup environment and run data pipelines
• [Reference Video Pipelines Guide](docs/curator/reference/video-pipelines.md) - details for general video processing pipelines
• [Reference AV Pipelines Guide](docs/curator/reference/av-pipelines.md) - details for multi-camera video, and (upcoming) GPS & LiDAR processing pipelines for autonomous vehicle (AV)
• [NVCF Guide](docs/client/nvcf-guide.md) - deployment instruction on Nvidia Cloud Functions

Developer Documentation

• [Developer Guide](docs/developer-guide.md) - information for contributors
• [Architecture Guide](docs/curator/reference/architecture.md) - diagrams and description to help understand core architecture
• [Pipeline Design Guide](docs/curator/guides/pipeline-design.md) - detailed walk-through of the hello-world pipeline and performance optimization points
• [Observability Guide](docs/curator/guides/observability.md) - instructions to setup and understand monitoring dashboard

AI Agent Context Files

• [AGENTS.md](AGENTS.md) - Context file for Codex
• [CLAUDE.md](CLAUDE.md) - Context file for Claude Code
• [GEMINI.md](GEMINI.md) - Context file for Gemini

Directory Structure

cosmos-curator/
├── cosmos_curator/ # Curator implementation
│ ├── client # CLI to run locally
│ ├── image_cli # Docker image management
│ ├── local_cli # Launch pipelines by running local container
│ ├── nvcf_cli # Launch pipelines on NVIDIA cloud function
│ ├── slurm_cli # Launch pipelines on Slurm cluster
│ ├── utils # Common utilities for various CLI apps
│ ├── core/ # Core functionality
│ ├── cf # Service entry point for a cloud function deployment
│ ├── interfaces # Core base class to integrate model and define new pipelines
│ ├── managers # CLIs to run inside the container to manage models, databases, etc.
│ ├── utils # Common utilities for pipelines
│ ├── models/ # AI model inference implementations
│ ├── pipelines/ # Pipeline implementations
│ ├── examples/ # Minimal example pipelines to help understand the framework
│ ├── video/ # Reference pipelines for video curation
│ ├── scripts/ # Startup scripts in various deployment environments
├── cosmos-xenna # Git submodule for https://github.com/nvidia-cosmos/cosmos-xenna
├── packages # Dockerfiles and scripts related to packaging
│ ├── cosmos_curator # Dockerfile template and Pixi environment post-install scripts
├── tests # Tests for testing
│ ├── cosmos_curator
│ ├── pipelines # Tests for models and pipeline stages for cosmos_curator
│ ├── client # Tests for client CLIs
├── examples # Example configuration files and scripts

Note: To initialize and update the cosmos-xenna submodule, run:

git submodule update --init --recursive

This ensures all submodule content is checked out correctly.

Support

For support and questions:

• Check the [documentation](docs/README.md)
• Open an issue on GitHub

Acknowledgments

• cosmos-xenna team for the core library
• All contributors and users of the project

Responsible Use of AI Models

[Responsible Use](./RESPONSIBLE_USE.md)

Notice and Disclaimer

NOTICE AND DISCLAIMER: This software automatically retrieves, accesses or interacts with external materials. Those retrieved materials are not distributed with this software and are governed solely by separate terms, conditions and licenses. You are solely responsible for finding, reviewing and complying with all applicable terms, conditions, and licenses, and for verifying the security, integrity and suitability of any retrieved materials for your specific use case. This software is provided "AS IS", without warranty of any kind. The author makes no representations or warranties regarding any retrieved materials, and assumes no liability for any losses, damages, liabilities or legal consequences from your use or inability to use this software or any retrieved materials. Use this software and the retrieved materials at your own risk.

License and Contact

This project will download and install additional third-party open source software projects. Review the license terms of these open source projects before use.

NVIDIA Cosmos source code is released under the Apache 2 License.

NVIDIA Cosmos models are released under the NVIDIA Open Model License. For a custom license, please contact…