segmentation_models_pytorch/docs/models.rst

📦 Segmentation Models
==============================

Unet
~~~~
.. autoclass:: segmentation_models_pytorch.Unet

Unet++
~~~~~~
.. autoclass:: segmentation_models_pytorch.UnetPlusPlus

EfficientUNet++
~~~~~~~~~~~~~~~
.. autoclass:: segmentation_models_pytorch.EfficientUnetPlusPlus

ResUnet
~~~~~~~
.. autoclass:: segmentation_models_pytorch.ResUnet

ResUnet++
~~~~~~~~~
.. autoclass:: segmentation_models_pytorch.ResUnetPlusPlus

MAnet
~~~~~~
.. autoclass:: segmentation_models_pytorch.MAnet

Linknet
~~~~~~~
.. autoclass:: segmentation_models_pytorch.Linknet

FPN
~~~
.. autoclass:: segmentation_models_pytorch.FPN

PSPNet
~~~~~~
.. autoclass:: segmentation_models_pytorch.PSPNet

PAN
~~~
.. autoclass:: segmentation_models_pytorch.PAN

DeepLabV3
~~~~~~~~~
.. autoclass:: segmentation_models_pytorch.DeepLabV3

DeepLabV3+
~~~~~~~~~~
.. autoclass:: segmentation_models_pytorch.DeepLabV3Plus