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. |