IAM datasets implemented.

1 files changed, 15 insertions, 2 deletions

diff --git a/src/text_recognizer/networks/__init__.py b/src/text_recognizer/networks/__init__.py
index a83ca35..d20c86a 100644
--- a/src/text_recognizer/networks/__init__.py
+++ b/src/text_recognizer/networks/__init__.py
@@ -1,6 +1,19 @@
 """Network modules."""
+from .ctc import greedy_decoder
 from .lenet import LeNet
+from .line_lstm_ctc import LineRecurrentNetwork
+from .misc import sliding_window
 from .mlp import MLP
-from .residual_network import ResidualNetwork
+from .residual_network import ResidualNetwork, ResidualNetworkEncoder
+from .wide_resnet import WideResidualNetwork
 
-__all__ = ["MLP", "LeNet", "ResidualNetwork"]
+__all__ = [
+    "greedy_decoder",
+    "MLP",
+    "LeNet",
+    "LineRecurrentNetwork",
+    "ResidualNetwork",
+    "ResidualNetworkEncoder",
+    "sliding_window",
+    "WideResidualNetwork",
+]