models.rst

🕸️ Segmentation Models

• Unet
• Unet++
• FPN
• PSPNet
• DeepLabV3
• DeepLabV3+
• Linknet
• MAnet
• PAN
• UPerNet
• Segformer
• DPT

Unet

.. autoclass:: segmentation_models_pytorch.Unet

Unet++

.. autoclass:: segmentation_models_pytorch.UnetPlusPlus

FPN

.. autoclass:: segmentation_models_pytorch.FPN

PSPNet

.. autoclass:: segmentation_models_pytorch.PSPNet

DeepLabV3

.. autoclass:: segmentation_models_pytorch.DeepLabV3

DeepLabV3+

.. autoclass:: segmentation_models_pytorch.DeepLabV3Plus

Linknet

.. autoclass:: segmentation_models_pytorch.Linknet

MAnet

.. autoclass:: segmentation_models_pytorch.MAnet

PAN

.. autoclass:: segmentation_models_pytorch.PAN

UPerNet

.. autoclass:: segmentation_models_pytorch.UPerNet

Segformer

.. autoclass:: segmentation_models_pytorch.Segformer

DPT

Note

See full list of DPT-compatible timm encoders in :ref:`dpt-encoders`.

Note

For some encoders, the model requires dynamic_img_size=True to be passed in order to work with resolutions different from what the encoder was trained for.

.. autoclass:: segmentation_models_pytorch.DPT