:github_url: https://github.com/taipingeric/fusionlab

.. _segmentation:

Segmentation Model
==============


.. automodule:: fusionlab.segmentation
.. currentmodule:: fusionlab.segmentation

`UNet`
~~~~~~~~~~

.. autoclass:: UNet
    :members:
.. autoclass:: TFUNet
    :members:

`ResUNet`
~~~~~~~~~~

.. autoclass:: ResUNet
    :members:
.. autoclass:: TFResUNet
    :members:

`UNet++`
~~~~~~~~~~

.. autoclass:: UNet2plus
    :members:
.. autoclass:: TFUNet2plus
    :members:

`TransUNet`
~~~~~~~~~~

.. autoclass:: TransUNet
    :members:

`UNETR`
~~~~~~~~~~

.. autoclass:: UNETR
    :members:

`SegFormer`
~~~~~~~~~~

.. autoclass:: SegFormer
    :members:

`HuggingFace Segmentation Model`
~~~~~~~~~~

.. autoclass:: HFSegmentationModel
    :members: