summaryrefslogtreecommitdiff
path: root/src/text_recognizer/models/__init__.py
blob: 28aa52eb4bdefe48a5f0f4aedb1e529b9c239df6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
"""Model modules."""
from .base import Model
from .character_model import CharacterModel
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",
    "accuracy",
    "TransformerEncoderModel",
    "VisionTransformerModel",
    "wer",
]