sleap-nn 0.0.2

Neural network backend for training and inference for animal pose estimation.

sleap-nn

Neural network backend for training and inference for animal pose estimation.

This is the deep learning engine that powers SLEAP (Social LEAP Estimates Animal Poses), providing neural network architectures for multi-instance animal pose estimation and tracking. Built on PyTorch, SLEAP-NN offers an end-to-end training workflow, supporting multiple model types (Single Instance, Top-Down, Bottom-Up, Multi-Class), and seamless integration with SLEAP's GUI and command-line tools.

Need a quick start? Refer to our Quick Start guide in the docs.

Documentation

📚 Documentation - Comprehensive guides and API reference

For development setup

1. Clone the sleap-nn repo

git clone https://github.com/talmolab/sleap-nn.git
cd sleap-nn

2. Install uv Install uv first - a fast Python package manager:

# macOS/Linux
curl -LsSf https://astral.sh/uv/install.sh | sh

# Windows
powershell -c "irm https://astral.sh/uv/install.ps1 | iex"

3. Install sleap-nn dependencies based on your platform\

• Sync all dependencies based on your correct wheel using uv sync. uv sync creates a .venv (virtual environment) inside your current working directory. This environment is only active within that directory and can't be directly accessed from outside. To use all installed packages, you must run commands with uv run (e.g., uv run sleap-nn train ... or uv run pytest ...).

  • Windows/Linux with NVIDIA GPU (CUDA 11.8):

  uv sync --extra dev --extra torch-cuda118
  

  • Windows/Linux with NVIDIA GPU (CUDA 12.8):

  uv sync --extra dev --extra torch-cuda128
  

  • macOS with Apple Silicon (M1, M2, M3, M4) or CPU-only (no GPU or unsupported GPU): Note: Even if torch-cpu is used on macOS, the MPS backend will be available.

  uv sync --extra dev --extra torch-cpu
  

4. Run tests

   uv run pytest tests
   

5. (Optional) Lint and format code

   uv run black --check sleap_nn tests
   uv run ruff check sleap_nn/