From e1b504bca41a9793ed7e88ef14f2e2cbd85724f2 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Tue, 8 Sep 2020 23:14:23 +0200
Subject: IAM datasets implemented.

---
 src/text_recognizer/datasets/__init__.py | 12 ++++++++++--
 1 file changed, 10 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 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",
 ]
-- 
cgit v1.2.3-70-g09d2