summaryrefslogtreecommitdiff
path: root/src/text_recognizer/datasets/__init__.py
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-08-05 01:03:37 +0200
committeraktersnurra <gustaf.rydholm@gmail.com>2020-08-05 01:03:37 +0200
commit125d5da5fb845d03bda91426e172bca7f537584a (patch)
tree6daf305555b76338ae482e81da58aa444a8255df /src/text_recognizer/datasets/__init__.py
parent0d0540952f79437026fc5a146b81e4b45190ff6a (diff)
Emnist lines data loader implemented.
Diffstat (limited to 'src/text_recognizer/datasets/__init__.py')
-rw-r--r--src/text_recognizer/datasets/__init__.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index a8c46c4..1b4cc59 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -1,21 +1,28 @@
"""Dataset modules."""
from .emnist_dataset import (
+ _augment_emnist_mapping,
+ _load_emnist_essentials,
DATA_DIRNAME,
EmnistDataLoaders,
EmnistDataset,
+ ESSENTIALS_FILENAME,
)
from .emnist_lines_dataset import (
construct_image_from_string,
+ EmnistLinesDataLoaders,
EmnistLinesDataset,
get_samples_by_character,
)
from .util import Transpose
__all__ = [
+ "_augment_emnist_mapping",
+ "_load_emnist_essentials",
"construct_image_from_string",
"DATA_DIRNAME",
"EmnistDataset",
"EmnistDataLoaders",
+ "EmnistLinesDataLoaders",
"EmnistLinesDataset",
"get_samples_by_character",
"Transpose",