Skip to content

architectures

sleap_nn.architectures

Modules related to model architectures.

Modules:

Name Description
common

Common utilities for architecture and model building.

convnext

This module provides a generalized implementation of ConvNext.

encoder_decoder

Generic encoder-decoder fully convolutional backbones.

heads

Model head definitions for defining model output types.

model

This module defines the main SLEAP model class for defining a trainable model.

swint

This module provides a generalized implementation of SwinT.

unet

This module provides a generalized implementation of UNet.

utils

Miscellaneous utility functions for architectures and modeling.