diff options
Diffstat (limited to 'src/text_recognizer/networks/__init__.py')
-rw-r--r-- | src/text_recognizer/networks/__init__.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/text_recognizer/networks/__init__.py b/src/text_recognizer/networks/__init__.py index 2b624bb..bac5d28 100644 --- a/src/text_recognizer/networks/__init__.py +++ b/src/text_recognizer/networks/__init__.py @@ -1,4 +1,5 @@ """Network modules.""" +from .cnn import CNN from .cnn_transformer import CNNTransformer from .crnn import ConvolutionalRecurrentNetwork from .ctc import greedy_decoder @@ -7,15 +8,19 @@ from .lenet import LeNet from .metrics import accuracy, cer, wer from .mlp import MLP from .residual_network import ResidualNetwork, ResidualNetworkEncoder +from .transducer import TDS2d from .transformer import Transformer from .unet import UNet from .util import sliding_window from .vit import ViT +from .vq_transformer import VQTransformer +from .vqvae import VQVAE from .wide_resnet import WideResidualNetwork __all__ = [ "accuracy", "cer", + "CNN", "CNNTransformer", "ConvolutionalRecurrentNetwork", "DenseNet", @@ -27,8 +32,11 @@ __all__ = [ "ResidualNetworkEncoder", "sliding_window", "UNet", + "TDS2d", "Transformer", "ViT", + "VQTransformer", + "VQVAE", "wer", "WideResidualNetwork", ] |