Metadata-Version: 2.3
Name: lightning-pose
Version: 1.9.1
Summary: Semi-supervised pose estimation using pytorch lightning
License: MIT
Keywords: machine learning,deep learning,computer_vision
Author: Dan Biderman
Author-email: danbider@gmail.com
Maintainer: Karan Sikka
Maintainer-email: karanssikka@gmail.com
Requires-Python: >=3.10
Classifier: Development Status :: 5 - Production/Stable
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Image Processing
Classifier: Topic :: Scientific/Engineering :: Visualization
Provides-Extra: dev
Provides-Extra: extra-models
Requires-Dist: Sphinx ; extra == "dev"
Requires-Dist: black ; extra == "dev"
Requires-Dist: fiftyone
Requires-Dist: flake8 ; extra == "dev"
Requires-Dist: h5py
Requires-Dist: hydra-core
Requires-Dist: imgaug
Requires-Dist: isort ; extra == "dev"
Requires-Dist: kaleido
Requires-Dist: kornia
Requires-Dist: lightning
Requires-Dist: lightning-bolts ; extra == "extra-models"
Requires-Dist: matplotlib
Requires-Dist: moviepy (<2.0.0)
Requires-Dist: numpy (<2.0.0)
Requires-Dist: nvidia-dali-cuda110
Requires-Dist: opencv-python-headless
Requires-Dist: pandas (>=2.0.0)
Requires-Dist: pillow
Requires-Dist: plotly
Requires-Dist: pytest ; extra == "dev"
Requires-Dist: pytest-html ; extra == "dev"
Requires-Dist: pytest-mock ; extra == "dev"
Requires-Dist: scikit-learn (>=1.5.0)
Requires-Dist: seaborn
Requires-Dist: sphinx-automodapi ; extra == "dev"
Requires-Dist: sphinx-copybutton ; extra == "dev"
Requires-Dist: sphinx-design ; extra == "dev"
Requires-Dist: sphinx-rtd-dark-mode ; extra == "dev"
Requires-Dist: sphinx_rtd_theme ; extra == "dev"
Requires-Dist: streamlit
Requires-Dist: tbparse (<1.0.0) ; extra == "dev"
Requires-Dist: tensorboard (<3.0.0)
Requires-Dist: torchtyping
Requires-Dist: torchvision
Requires-Dist: transformers
Requires-Dist: typeguard
Project-URL: Documentation, https://lightning-pose.readthedocs.io/
Project-URL: Repository, https://github.com/paninski-lab/lightning-pose
Description-Content-Type: text/markdown

![](https://github.com/paninski-lab/lightning-pose/raw/main/docs/images/LightningPose_horizontal_light.png)

[![Discord](https://img.shields.io/discord/1103381776895856720)](https://discord.gg/tDUPdRj4BM)
![GitHub](https://img.shields.io/github/license/paninski-lab/lightning-pose)
[![Documentation Status](https://readthedocs.org/projects/lightning-pose/badge/?version=latest)](https://lightning-pose.readthedocs.io/en/latest/?badge=latest)
![PyPI](https://img.shields.io/pypi/v/lightning-pose)
![PyPI Downloads](https://static.pepy.tech/badge/lightning-pose/week)


Pose estimation models implemented in **Pytorch Lightning**, supporting massively accelerated training on _unlabeled_ videos using **NVIDIA DALI**. 
Models can be evaluated with **TensorBoard**, **FiftyOne**, and **Streamlit**.

As of June 2024, Lightning Pose is now [published in Nature Methods](https://rdcu.be/dLP3z)!

## Try our demo

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/paninski-lab/lightning-pose/blob/main/scripts/litpose_training_demo.ipynb)

Train a network on an example dataset and visualize the results in Google Colab.

## Getting Started
Please see the [Lightning Pose documentation](https://lightning-pose.readthedocs.io/) 
for installation instructions and user guides.
Note that the Lightning Pose package provides tools for training and evaluating models on 
_already labeled data_ and unlabeled video clips. 

We also offer a [browser-based application](https://github.com/Lightning-Universe/Pose-app) that 
supports the full life cycle of a pose estimation project, from data annotation to model training 
to diagnostic visualizations.

The Lightning Pose team also actively develops the 
[Ensemble Kalman Smoother (EKS)](https://github.com/paninski-lab/eks), 
a simple and performant post-processor that works with any pose estimation package including 
Lightning Pose, DeepLabCut, and SLEAP.

## Community

Lightning Pose is primarily maintained by 
[Karan Sikka](https://github.com/ksikka) (Columbia University),
[Matt Whiteway](https://themattinthehatt.github.io) (Columbia University),
and
[Dan Biderman](https://dan-biderman.netlify.app) (Stanford University). 
 

Lightning Pose is under active development and we welcome community contributions.
Whether you want to implement some of your own ideas or help out with our [development roadmap](docs/roadmap.md), please get in touch with us on Discord (see contributing guidelines [here](CONTRIBUTING.md)). 

