Inital commit for refactoring to lightning

1 files changed, 39 insertions, 0 deletions

diff --git a/text_recognizer/datasets/__init__.py b/text_recognizer/datasets/__init__.py
new file mode 100644
index 0000000..a6c1c59
--- /dev/null
+++ b/text_recognizer/datasets/__init__.py
@@ -0,0 +1,39 @@
+"""Dataset modules."""
+from .emnist_dataset import EmnistDataset
+from .emnist_lines_dataset import (
+    construct_image_from_string,
+    EmnistLinesDataset,
+    get_samples_by_character,
+)
+from .iam_dataset import IamDataset
+from .iam_lines_dataset import IamLinesDataset
+from .iam_paragraphs_dataset import IamParagraphsDataset
+from .iam_preprocessor import load_metadata, Preprocessor
+from .transforms import AddTokens, Transpose
+from .util import (
+    _download_raw_dataset,
+    compute_sha256,
+    DATA_DIRNAME,
+    download_url,
+    EmnistMapper,
+    ESSENTIALS_FILENAME,
+)
+
+__all__ = [
+    "_download_raw_dataset",
+    "AddTokens",
+    "compute_sha256",
+    "construct_image_from_string",
+    "DATA_DIRNAME",
+    "download_url",
+    "EmnistDataset",
+    "EmnistMapper",
+    "EmnistLinesDataset",
+    "get_samples_by_character",
+    "load_metadata",
+    "IamDataset",
+    "IamLinesDataset",
+    "IamParagraphsDataset",
+    "Preprocessor",
+    "Transpose",
+]