summaryrefslogtreecommitdiff
path: root/src/text_recognizer/models/__init__.py
blob: 53340f18c77bc66b50b081602c3c51d0523f585d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
"""Model modules."""
from .base import Model
from .character_model import CharacterModel
from .crnn_model import CRNNModel
from .metrics import accuracy, accuracy_ignore_pad, cer, wer
from .transformer_model import TransformerModel

__all__ = [
    "accuracy",
    "accuracy_ignore_pad",
    "cer",
    "CharacterModel",
    "CRNNModel",
    "Model",
    "TransformerModel",
    "wer",
]