Inital commit for refactoring to lightning

1 files changed, 43 insertions, 0 deletions

diff --git a/text_recognizer/networks/__init__.py b/text_recognizer/networks/__init__.py
new file mode 100644
index 0000000..1521355
--- /dev/null
+++ b/text_recognizer/networks/__init__.py
@@ -0,0 +1,43 @@
+"""Network modules."""
+from .cnn import CNN
+from .cnn_transformer import CNNTransformer
+from .crnn import ConvolutionalRecurrentNetwork
+from .ctc import greedy_decoder
+from .densenet import DenseNet
+from .lenet import LeNet
+from .metrics import accuracy, cer, wer
+from .mlp import MLP
+from .residual_network import ResidualNetwork, ResidualNetworkEncoder
+from .transducer import load_transducer_loss, 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",
+    "FCN",
+    "greedy_decoder",
+    "MLP",
+    "LeNet",
+    "load_transducer_loss",
+    "ResidualNetwork",
+    "ResidualNetworkEncoder",
+    "sliding_window",
+    "UNet",
+    "TDS2d",
+    "Transformer",
+    "ViT",
+    "VQTransformer",
+    "VQVAE",
+    "wer",
+    "WideResidualNetwork",
+]