diff --git a/README.md b/README.md index e0a3b42..894f870 100644 --- a/README.md +++ b/README.md @@ -1,13 +1,9 @@ [![Generic badge](https://img.shields.io/badge/Contributions-Welcome-brightgreen.svg)](README.md) Code style: black[![License: LGPL v3](https://img.shields.io/badge/License-LGPL%20v3-blue.svg)](https://www.gnu.org/licenses/lgpl-3.0) -# DLClibrary +# DLClibrary: the API interface for DeepLabCut ModelZoo models on Hugging Face -DLClibrary is a lightweight library supporting universal functions for the [DeepLabCut](https://github.com/DeepLabCut/DeepLabCut) ecosystem. - -Supported functions (at this point): - -- API for downloading model weights from [the model zoo](http://www.mackenziemathislab.org/dlc-modelzoo) +DLClibrary is a lightweight library supporting an API for downloading model weights from [the model zoo](http://www.mackenziemathislab.org/dlc-modelzoo) # Quick start @@ -75,13 +71,13 @@ https://github.com/DeepLabCut/DLClibrary/blob/main/dlclibrary/dlcmodelzoo/modelz ```yaml my_cool_dataset: # name of the dataset used to train the model detectors: - detector_name: path/to/huggingface-detector.pt # add detectors under `detector` + detector_name: path/to/huggingface-detector.pt # add detectors under `detectors` pose_models: pose_model_name: path/to/huggingface-pose-model.pt # add pose models under `pose_models` other_pose_model_name: path/to/huggingface-other-pose-model.pt ``` -This will allow users to download the models using the format `datatsetName_modelName`, +This will allow users to download the models using the format `datasetName_modelName`, i.e. for this example 3 models would be available: `my_cool_dataset_detector_name`, `my_cool_dataset_pose_model_name` and `my_cool_dataset_other_pose_model_name`. @@ -93,3 +89,22 @@ To add a new model for `deeplabcut >= 3.0.0`, simply: The models will then be listed when calling `dlclibrary.get_available_detectors` or `dlclibrary.get_available_models`! You can list the datasets for which models are available using `dlclibrary.get_available_datasets`. + +## Citation + +If you use SuperAnimal, please cite: + +```bibtex +@article{ye2024superanimal, + title = {SuperAnimal pretrained pose estimation models for behavioral analysis}, + author = {Ye, Shaokai and Filippova, Anastasiia and Lauer, Jessy and Schneider, Steffen and Vidal, Maxime and Qiu, Tian and Mathis, Alexander and Mathis, Mackenzie Weygandt}, + journal = {Nature Communications}, + volume = {15}, + number = {1}, + pages = {5165}, + year = {2024}, + doi = {10.1038/s41467-024-48792-2} +} +``` + +