Skip to main content

Convert DeepLabCut dataset to YOLO format

Project description

deeplabcut2yolo

Convert DLC to YOLO,
Lightning-fast and hassle-free.

License: GPL v3 License: BSD-3-Clause PyPI Package Version Package Total Downloads Documentation Cite

deeplabcut2yolo facilitates training DeepLabCut datasets on YOLO models. Deeplabcut2yolo automatically converts DeepLabCut (DLC) labels to COCO-like format compatible with YOLO, while providing customizability for more advanced users, so you can spend your energy on what matters!

Results from d2y All DeepLabCut datasets belong to their respective owner under CC BY-NC 4.0. This particular image is the training data for YOLO, converted using deeplabcut2yolo from the Tri-Mouse dataset (Lauer et al., 2022).

Quick Start

import deeplabcut2yolo as d2y

# In its simplest form,
d2y.convert("./deeplabcut-dataset/")

# To also generate data.yml
d2y.convert(
    dataset_path,
    train_paths=train_paths,
    val_paths=val_paths,
    skeleton_symmetric_pairs=skeleton_symmetric_pairs,
    data_yml_path="data.yml",
    class_names=class_names,
    verbose=True,
)

To install deeplabcut2yolo using pip:

pip install deeplabcut2yolo

For more information, see examples and documentation.

Features

  • Automatically detect default DeepLabCut dataset structure
  • Vectorized label conversion
  • Support single- and multi-animal projects
  • Convenient data.yml generation function for YOLO models

Contribution

You can contribute to deeplabcut2yolo by making pull requests. Currently, these are high-priority features:

  • Testing module and test cases
  • Documentation

Citation

Citation is not required but is greatly appreciated. If this project helps you, please cite using the following APA-style reference:

Pornsiriprasert, S. (2025). Deeplabcut2yolo: A DeepLabCut-to-YOLO Dataset Converter for Python (v2.2.7). Zenodo. https://doi.org/10.5281/zenodo.17386187

or this BibTeX entry.

@software{pornsiriprasert2025,
  author       = {Pornsiriprasert, Sira},
  title        = {Deeplabcut2yolo: A DeepLabCut-to-YOLO Dataset Converter for Python},
  month        = dec,
  year         = 2025,
  publisher    = {Zenodo},
  version      = {v2.2.7},
  doi          = {10.5281/zenodo.17386187},
  url          = {https://doi.org/10.5281/zenodo.17386187},
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

deeplabcut2yolo-2.2.8.tar.gz (10.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

deeplabcut2yolo-2.2.8-py3-none-any.whl (10.0 kB view details)

Uploaded Python 3

File details

Details for the file deeplabcut2yolo-2.2.8.tar.gz.

File metadata

  • Download URL: deeplabcut2yolo-2.2.8.tar.gz
  • Upload date:
  • Size: 10.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for deeplabcut2yolo-2.2.8.tar.gz
Algorithm Hash digest
SHA256 7e6e20a450ca873eebb97b099f6d2dceec35deeaf6669d5c1cdb30923572a5dc
MD5 1a14a21933f269ee49fbd2202fe08025
BLAKE2b-256 b9c7726b1b666243c05ed2e29bbb96c44ec2cac4fce0e74f8af9fe35b286a55b

See more details on using hashes here.

Provenance

The following attestation bundles were made for deeplabcut2yolo-2.2.8.tar.gz:

Publisher: python-publish.yml on p-sira/deeplabcut2yolo

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file deeplabcut2yolo-2.2.8-py3-none-any.whl.

File metadata

File hashes

Hashes for deeplabcut2yolo-2.2.8-py3-none-any.whl
Algorithm Hash digest
SHA256 19509e36ca819d1559bd32691675fed887eb0742bb5f53d7a86d050cfa03536b
MD5 2f2ebdfed0a373b7c6d00aaa66f4e501
BLAKE2b-256 641abb332f424d479bb087143c2f8bf7aef4dfccf1d05bd7d85f390a582cc3db

See more details on using hashes here.

Provenance

The following attestation bundles were made for deeplabcut2yolo-2.2.8-py3-none-any.whl:

Publisher: python-publish.yml on p-sira/deeplabcut2yolo

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page