summaryrefslogtreecommitdiff
path: root/src/text_recognizer/datasets/__init__.py
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-08-03 23:33:34 +0200
committeraktersnurra <gustaf.rydholm@gmail.com>2020-08-03 23:33:34 +0200
commit07dd14116fe1d8148fb614b160245287533620fc (patch)
tree63395d88b17a14ad453c52889fcf541e6cbbdd3e /src/text_recognizer/datasets/__init__.py
parent704451318eb6b0b600ab314cb5aabfac82416bda (diff)
Working Emnist lines dataset.
Diffstat (limited to 'src/text_recognizer/datasets/__init__.py')
-rw-r--r--src/text_recognizer/datasets/__init__.py24
1 files changed, 22 insertions, 2 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index 795be90..bfa6a02 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -1,4 +1,24 @@
"""Dataset modules."""
-from .emnist_dataset import EmnistDataLoader
+from .emnist_dataset import (
+ DATA_DIRNAME,
+ EmnistDataLoaders,
+ EmnistDataset,
+)
+from .emnist_lines_dataset import (
+ construct_image_from_string,
+ EmnistLinesDataset,
+ get_samples_by_character,
+)
+from .sentence_generator import SentenceGenerator
+from .util import Transpose
-__all__ = ["EmnistDataLoader"]
+__all__ = [
+ "construct_image_from_string",
+ "DATA_DIRNAME",
+ "EmnistDataset",
+ "EmnistDataLoaders",
+ "EmnistLinesDataset",
+ "get_samples_by_character",
+ "SentenceGenerator",
+ "Transpose",
+]