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 PyPI Package Version Package Total Downloads Documentation

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 Python Library for Converting DeepLabCut Dataset to YOLO Format (Version 2.2.5) [Computer software]. GitHub. https://github.com/p-sira/deeplabcut2yolo/

or this BibTeX entry.

@software{deeplabcut2yolo,
    author = {{Pornsiriprasert, S}},
    title = {Deeplabcut2yolo: A Python Library for Converting DeepLabCut Dataset to YOLO Format},
    url = {https://github.com/p-sira/deeplabcut2yolo/},
    version = {2.2.5},
    publisher = {GitHub},
    year = {2025},
    month = {5},
}

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.5.tar.gz (9.3 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.5-py3-none-any.whl (9.4 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for deeplabcut2yolo-2.2.5.tar.gz
Algorithm Hash digest
SHA256 5485991c9c8901a605fe67912edbf6eb61ca36036dfcb0f8e2d1ada480e2124c
MD5 022b0c46cbf5fbdb6ff6d1674b49d7cd
BLAKE2b-256 ac330a42a8184c1b35945bdbd4d9c8116c1c9a83e791ff00e6947d600dc8fa9c

See more details on using hashes here.

Provenance

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

File metadata

File hashes

Hashes for deeplabcut2yolo-2.2.5-py3-none-any.whl
Algorithm Hash digest
SHA256 97aabf739863e65986ffdc931cbce82b06ffa2d56ebded17bc54933d41fc48a7
MD5 7506aa60ea9236b6d9228e44a6b6d6ee
BLAKE2b-256 ee83a88171c957897c8395d804be0267f59081bdf8068e5e6f6eb7f4dc93f92b

See more details on using hashes here.

Provenance

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