From 07dd14116fe1d8148fb614b160245287533620fc Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Mon, 3 Aug 2020 23:33:34 +0200
Subject: Working Emnist lines dataset.

---
 src/text_recognizer/datasets/__init__.py | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

(limited to 'src/text_recognizer/datasets/__init__.py')

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",
+]
-- 
cgit v1.2.3-70-g09d2