From 7e8e54e84c63171e748bbf09516fd517e6821ace Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 20 Mar 2021 18:09:06 +0100 Subject: Inital commit for refactoring to lightning --- text_recognizer/datasets/__init__.py | 39 ++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 text_recognizer/datasets/__init__.py (limited to 'text_recognizer/datasets/__init__.py') 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", +] -- cgit v1.2.3-70-g09d2