sunnychenxiwang's picture
Upload 1600 files
14c9181 verified
.. role:: hidden
:class: hidden-section
mmocr.models
===================================
- :mod:`~mmocr.models.common`
- :ref:`commombackbones`
- :ref:`commomdictionary`
- :ref:`commomlayers`
- :ref:`commomlosses`
- :ref:`commommodules`
- :mod:`~mmocr.models.textdet`
- :ref:`detdetectors`
- :ref:`detdatapreprocessors`
- :ref:`detnecks`
- :ref:`detheads`
- :ref:`detmodulelosses`
- :ref:`detpostprocessors`
- :mod:`~mmocr.models.textrecog`
- :ref:`recrecognizers`
- :ref:`recdatapreprocessors`
- :ref:`recpreprocessors`
- :ref:`recencoders`
- :ref:`recdecoders`
- :ref:`recmodulelosses`
- :ref:`recpostprocessors`
- :ref:`reclayers`
- :mod:`~mmocr.models.kie`
- :ref:`kieextractors`
- :ref:`kieheads`
- :ref:`kiemodulelosses`
- :ref:`kiepostprocessors`
.. module:: mmocr.models.common
models.common
---------------------------------------------
.. currentmodule:: mmocr.models.common
.. _commombackbones:
BackBones
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
UNet
.. _commomdictionary:
Dictionary
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
Dictionary
.. _commomlosses:
Losses
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
MaskedBalancedBCEWithLogitsLoss
MaskedDiceLoss
MaskedSmoothL1Loss
MaskedSquareDiceLoss
MaskedBCEWithLogitsLoss
SmoothL1Loss
CrossEntropyLoss
MaskedBalancedBCELoss
MaskedBCELoss
.. _commomlayers:
Layers
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
TFEncoderLayer
TFDecoderLayer
.. _commommodules:
Modules
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
ScaledDotProductAttention
MultiHeadAttention
PositionwiseFeedForward
PositionalEncoding
.. module:: mmocr.models.textdet
models.textdet
---------------------------------------------
.. currentmodule:: mmocr.models.textdet
.. _detdetectors:
Detectors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SingleStageTextDetector
DBNet
PANet
PSENet
TextSnake
FCENet
DRRG
MMDetWrapper
.. _detdatapreprocessors:
Data Preprocessors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
TextDetDataPreprocessor
.. _detnecks:
Necks
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
FPEM_FFM
FPNF
FPNC
FPN_UNet
.. _detheads:
Heads
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BaseTextDetHead
PSEHead
PANHead
DBHead
FCEHead
TextSnakeHead
DRRGHead
.. _detmodulelosses:
Module Losses
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SegBasedModuleLoss
PANModuleLoss
PSEModuleLoss
DBModuleLoss
TextSnakeModuleLoss
FCEModuleLoss
DRRGModuleLoss
.. _detpostprocessors:
Postprocessors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BaseTextDetPostProcessor
PSEPostprocessor
PANPostprocessor
DBPostprocessor
DRRGPostprocessor
FCEPostprocessor
TextSnakePostprocessor
.. module:: mmocr.models.textrecog
models.textrecog
---------------------------------------------
.. currentmodule:: mmocr.models.textrecog
.. _recrecognizers:
Recognizers
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BaseRecognizer
EncoderDecoderRecognizer
CRNN
SARNet
NRTR
RobustScanner
SATRN
ABINet
MASTER
ASTER
.. _recdatapreprocessors:
Data Preprocessors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
TextRecogDataPreprocessor
.. _recpreprocessors:
Preprocessors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
STN
.. _recbackbones:
BackBones
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
ResNet31OCR
MiniVGG
NRTRModalityTransform
ShallowCNN
ResNetABI
ResNet
MobileNetV2
.. _recencoders:
Encoders
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SAREncoder
NRTREncoder
BaseEncoder
ChannelReductionEncoder
SATRNEncoder
ABIEncoder
ASTEREncoder
.. _recdecoders:
Decoders
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BaseDecoder
ABILanguageDecoder
ABIVisionDecoder
ABIFuser
CRNNDecoder
ParallelSARDecoder
SequentialSARDecoder
ParallelSARDecoderWithBS
NRTRDecoder
SequenceAttentionDecoder
PositionAttentionDecoder
RobustScannerFuser
MasterDecoder
ASTERDecoder
.. _recmodulelosses:
Module Losses
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BaseTextRecogModuleLoss
CEModuleLoss
CTCModuleLoss
ABIModuleLoss
.. _recpostprocessors:
Postprocessors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BaseTextRecogPostprocessor
AttentionPostprocessor
CTCPostProcessor
.. _reclayers:
Layers
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
BidirectionalLSTM
Adaptive2DPositionalEncoding
BasicBlock
Bottleneck
RobustScannerFusionLayer
DotProductAttentionLayer
PositionAwareLayer
SATRNEncoderLayer
.. module:: mmocr.models.kie
models.kie
---------------------------------------------
.. currentmodule:: mmocr.models.kie
.. _kieextractors:
Extractors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SDMGR
.. _kieheads:
Heads
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SDMGRHead
.. _kiemodulelosses:
Module Losses
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SDMGRModuleLoss
.. _kiepostprocessors:
Postprocessors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autosummary::
:toctree: generated
:nosignatures:
:template: classtemplate.rst
SDMGRPostProcessor