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.6). Zenodo. https://doi.org/10.5281/zenodo.17386188

or this BibTeX entry.

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

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.6.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.6-py3-none-any.whl (10.0 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: deeplabcut2yolo-2.2.6.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.6.tar.gz
Algorithm Hash digest
SHA256 8c0b54aed238cc570a952aeb57d385708e3c5587a384e67012c1cde7d7a93fb2
MD5 1a7b95b690823521b514e5325bcc10fd
BLAKE2b-256 2ddaba5b2d6e35f8e876da82883def4ee5a4024e3809ec3512ae13d3010bc030

See more details on using hashes here.

Provenance

The following attestation bundles were made for deeplabcut2yolo-2.2.6.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.6-py3-none-any.whl.

File metadata

File hashes

Hashes for deeplabcut2yolo-2.2.6-py3-none-any.whl
Algorithm Hash digest
SHA256 73f6f8203011ee49ef66c4a5fdb06a55168cf447587f80c25a98efe3d8e4999c
MD5 a16bf2b01e7f2768df8de4ea5b7fb6b4
BLAKE2b-256 9d95c6c9f376b0fcd4e8afda3df48500bcb7a75bd4c72405e9e512e687f722c7

See more details on using hashes here.

Provenance

The following attestation bundles were made for deeplabcut2yolo-2.2.6-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