diff options
Diffstat (limited to 'text_recognizer/models/__init__.py')
| -rw-r--r-- | text_recognizer/models/__init__.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/text_recognizer/models/__init__.py b/text_recognizer/models/__init__.py new file mode 100644 index 0000000..7647d7e --- /dev/null +++ b/text_recognizer/models/__init__.py @@ -0,0 +1,18 @@ +"""Model modules.""" +from .base import Model +from .character_model import CharacterModel +from .crnn_model import CRNNModel +from .ctc_transformer_model import CTCTransformerModel +from .segmentation_model import SegmentationModel +from .transformer_model import TransformerModel +from .vqvae_model import VQVAEModel + +__all__ = [ + "CharacterModel", + "CRNNModel", + "CTCTransformerModel", + "Model", + "SegmentationModel", + "TransformerModel", + "VQVAEModel", +] |