Metadata-Version: 2.4
Name: viva-datacreator
Version: 0.0.7
Summary: An open-source tool for creating image segmentation datasets from videos using SAM 2
Author-email: "Miguel S. Soriano-García, Diego Mercado-Ravell, Israel Becerra, Julio De La Torre-Vanegas" <m16soriano@gmail.com>
License: MIT
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.12
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: pyyaml>=6.0.3
Requires-Dist: torch>=2.7.0
Requires-Dist: torchvision>=0.22.0
Requires-Dist: sam2
Requires-Dist: markupsafe<3.0
Requires-Dist: opencv-python>=4.11.0.86
Requires-Dist: matplotlib>=3.10.7
Requires-Dist: sahi>=0.11.36
Requires-Dist: pandas>=2.3.3
Requires-Dist: ipython>=9.7.0
Requires-Dist: ultralytics>=8.3.228
Requires-Dist: tqdm>=4.67.1
Requires-Dist: deep-sort-realtime>=1.3.2
Requires-Dist: setuptools<71
Requires-Dist: typer>=0.20.0
Provides-Extra: docs
Requires-Dist: mkdocs>=1.6.1; extra == "docs"
Requires-Dist: mkdocs-material>=9.6.17; extra == "docs"
Requires-Dist: mkdocstrings[python]>=0.26.1; extra == "docs"
Dynamic: license-file

# ViVa-DataCreator: An Open-Source Human-in-the-Loop Data Annotation Engine for Semantic Segmentation
[![PyPI version](https://badge.fury.io/py/viva-datacreator.svg)](https://badge.fury.io/py/viva-datacreator)
[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
[![Python versions](https://img.shields.io/badge/Python-3.12+-blue.svg)](https://www.python.org/)

<div align="center">
  <img
    src="https://github.com/user-attachments/assets/328d3d54-91af-4334-af12-6155a1e19718"
    width="600"
    alt="ViVa-DataCreator Logo"
  />
</div>

**ViVa-DataCreator** is an open-source tool for creating semantic segmentation datasets by tracking objects of interest from videos. It leverages the *Segment Anything Model 2 (SAM2)* and *You Only Look Once (YOLO)* AI models to perform segmentation and object detection, guiding users through an 8-step process to generate complete datasets ready for model training.

<figure style="text-align: center;">
  <img
    src="https://github.com/user-attachments/assets/5feba7c9-e32a-4f66-97fc-1a22f3f2d0c1"
    alt="ViVa-DataCreator GUI"
    width="800"
  />
  <figcaption>ViVa-DataCreator: Graphical User Interface for Dataset Creation</figcaption>
</figure>

This tool focuses on generating semantic segmentation datasets through object tracking, utilizing SAM 2 to enhance segmentation accuracy.

## Key Features

-   **Video-to-Dataset Conversion:** Transform videos into high-quality segmentation datasets with minimal manual effort.
-   **SAM 2 Integration:** Utilize the latest Segment Anything Model 2 for accurate and interactive segmentation.
-   **8-Step Flexible Process:** A comprehensive workflow that guides you through dataset creation, allowing you to move between steps as needed.
-   **Interactive Refinement:** Manually refine segmentations for precision and quality control.
-   **Object Tracking Integration:** Utilize YOLO and DeepSort for tracking objects of interest across video frames.
-   **Batch Processing:** Efficiently handle large videos through configurable batch processing.
-   **Customizable Classes:** Define and assign custom object classes with unique colors.
-   **Safety-Focused:** Designed for safe and reliable dataset generation without hardware risks.

## Documentation

For detailed usage instructions, examples, and API documentation, please refer to the [ViVa-DataCreator Documentation](https://viva-safeland.github.io/viva_datacreator/).



## Citation

If you use ViVa-DataCreator in your research, please consider adding the following citations:

**ViVa-DataCreator**
```bibtex
@software{soriano2025datacreator,
  author = {Miguel Soriano-García, Diego Mercado-Ravell, Israel Becerra and Julio De La Torre-Vanegas},
  title = {ViVa-DataCreator: An Open-Source Human-in-the-Loop Data Annotation Engine for Semantic Segmentation},
  year = {2025},
  url = {https://github.com/viva-safeland/viva_datacreator}
}
```

**ViVa-SAFELAND Simulator**
```bibtex
@article{soriano2025viva,
  title={ViVa-SAFELAND: a New Freeware for Safe Validation of Vision-based Navigation in Aerial Vehicles},
  author={Miguel S. Soriano-Garcia and Diego A. Mercado-Ravell},
  journal={arXiv preprint arXiv:2503.14719},
  year={2024}
}
```

**Related Application**
```bibtex
@misc{delatorre2025riskaware,
      title={Vision-Based Risk Aware Emergency Landing for UAVs in Complex Urban Environments}, 
      author={Julio de la Torre-Vanegas and Miguel Soriano-Garcia and Israel Becerra and Diego Mercado-Ravell},
      year={2025},
      eprint={2505.20423},
      archivePrefix={arXiv},
      primaryClass={cs.RO},
      url={https://arxiv.org/abs/2505.20423}, 
}
```
