diff options
Diffstat (limited to 'src/text_recognizer/datasets/__init__.py')
| -rw-r--r-- | src/text_recognizer/datasets/__init__.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py index 05f74f6..ede4541 100644 --- a/src/text_recognizer/datasets/__init__.py +++ b/src/text_recognizer/datasets/__init__.py @@ -10,15 +10,23 @@ from .emnist_lines_dataset import ( EmnistLinesDataset, get_samples_by_character, ) -from .util import fetch_data_loaders, Transpose +from .iam_dataset import IamDataset +from .iam_lines_dataset import IamLinesDataset +from .iam_paragraphs_dataset import IamParagraphsDataset +from .util import _download_raw_dataset, compute_sha256, download_url, Transpose __all__ = [ + "_download_raw_dataset", + "compute_sha256", "construct_image_from_string", "DATA_DIRNAME", + "download_url", "EmnistDataset", "EmnistMapper", "EmnistLinesDataset", - "fetch_data_loaders", "get_samples_by_character", + "IamDataset", + "IamLinesDataset", + "IamParagraphsDataset", "Transpose", ] |