diff options
Diffstat (limited to 'src/text_recognizer/models/__init__.py')
-rw-r--r-- | src/text_recognizer/models/__init__.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/text_recognizer/models/__init__.py b/src/text_recognizer/models/__init__.py index 0855079..28aa52e 100644 --- a/src/text_recognizer/models/__init__.py +++ b/src/text_recognizer/models/__init__.py @@ -1,16 +1,19 @@ """Model modules.""" from .base import Model from .character_model import CharacterModel -from .line_ctc_model import LineCTCModel +from .crnn_model import CRNNModel from .metrics import accuracy, cer, wer +from .transformer_encoder_model import TransformerEncoderModel from .vision_transformer_model import VisionTransformerModel __all__ = [ "Model", "cer", "CharacterModel", + "CRNNModel", "CNNTransfromerModel", - "LineCTCModel", "accuracy", + "TransformerEncoderModel", + "VisionTransformerModel", "wer", ] |