hat.models
===========

Models widely used in upper module in HAT.


Models
------


structures
^^^^^^^^^^^

.. py:currentmodule:: hat.models.structures

.. autosummary::
    :nosignatures:

    Classifier

    Segmentor

detectors
*********

.. py:currentmodule:: hat.models.structures.detectors

.. autosummary::
    :nosignatures:

    FCOS

backbones
^^^^^^^^^^

.. py:currentmodule:: hat.models.backbones

.. autosummary::
    :nosignatures:

    EfficientNet

    MobileNetV1

    ResNet18

    VargConvNet

    VargNetV2
    

necks
^^^^^^

.. py:currentmodule:: hat.models.necks

.. autosummary::
    :nosignatures:

    BiFPN

    DwUnet

    FPN

    RetinaNetFPN

    Unet

    PAFPN

    FastSCNNNeck

task_modules
^^^^^^^^^^^^

fcos
****

.. py:currentmodule:: hat.models.task_modules.fcos

.. autosummary::
    :nosignatures:

    FCOSDecoder

    FCOSHead

    FCOSTarget


seg
***

.. py:currentmodule:: hat.models.task_modules.seg

.. autosummary::
    :nosignatures:

    SegDecoder

    SegHead

    FRCNNSegHead

    SegTarget

fcn
****

.. py:currentmodule:: hat.models.task_modules.fcn

.. autosummary::
       :nosignatures:

    FCNHead
    
    DepthwiseSeparableFCNHead

    FCNDecoder

    FCNTarget


deeplab
*******

.. py:currentmodule:: hat.models.task_modules.deeplab

.. autosummary::
          :nosignatures:

    Deeplabv3plusHead
                                                                     


losses
^^^^^^^

.. py:currentmodule:: hat.models.losses

.. autosummary::
    :nosignatures:

    CEWithLabelSmooth

    CrossEntropyLoss

    CrossEntropyLossV2

    FocalLoss

    SoftmaxFocalLoss

    GIoULoss

    SegLoss

    SmoothL1Loss

    YOLOV3Loss

API Reference
-------------

.. automodule:: hat.models.structures
    :members:

.. automodule:: hat.models.structures.detectors
    :members:

.. automodule:: hat.models.backbones
    :members:

.. automodule:: hat.models.necks
    :members:

.. automodule:: hat.models.losses
    :members:

.. automodule:: hat.models.task_modules.fcos
    :members:


.. automodule:: hat.models.task_modules.seg
    :members: