src/text_recognizer/models/__init__.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

"""Model modules."""
from .base import Model
from .character_model import CharacterModel
from .line_ctc_model import LineCTCModel
from .metrics import accuracy, cer, wer
from .vision_transformer_model import VisionTransformerModel

__all__ = [
    "Model",
    "cer",
    "CharacterModel",
    "CNNTransfromerModel",
    "LineCTCModel",
    "accuracy",
    "wer",
]