summaryrefslogtreecommitdiff
path: root/src/text_recognizer/datasets/__init__.py
blob: a6c1c5996f674c04b19c0c45f66b0e99fd775e0d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
"""Dataset modules."""
from .emnist_dataset import EmnistDataset
from .emnist_lines_dataset import (
    construct_image_from_string,
    EmnistLinesDataset,
    get_samples_by_character,
)
from .iam_dataset import IamDataset
from .iam_lines_dataset import IamLinesDataset
from .iam_paragraphs_dataset import IamParagraphsDataset
from .iam_preprocessor import load_metadata, Preprocessor
from .transforms import AddTokens, Transpose
from .util import (
    _download_raw_dataset,
    compute_sha256,
    DATA_DIRNAME,
    download_url,
    EmnistMapper,
    ESSENTIALS_FILENAME,
)

__all__ = [
    "_download_raw_dataset",
    "AddTokens",
    "compute_sha256",
    "construct_image_from_string",
    "DATA_DIRNAME",
    "download_url",
    "EmnistDataset",
    "EmnistMapper",
    "EmnistLinesDataset",
    "get_samples_by_character",
    "load_metadata",
    "IamDataset",
    "IamLinesDataset",
    "IamParagraphsDataset",
    "Preprocessor",
    "Transpose",
]