summaryrefslogtreecommitdiff
path: root/src/text_recognizer
diff options
context:
space:
mode:
Diffstat (limited to 'src/text_recognizer')
-rw-r--r--src/text_recognizer/datasets/__init__.py2
-rw-r--r--src/text_recognizer/datasets/emnist_lines_dataset.py3
2 files changed, 2 insertions, 3 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index bfa6a02..a8c46c4 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -9,7 +9,6 @@ from .emnist_lines_dataset import (
EmnistLinesDataset,
get_samples_by_character,
)
-from .sentence_generator import SentenceGenerator
from .util import Transpose
__all__ = [
@@ -19,6 +18,5 @@ __all__ = [
"EmnistDataLoaders",
"EmnistLinesDataset",
"get_samples_by_character",
- "SentenceGenerator",
"Transpose",
]
diff --git a/src/text_recognizer/datasets/emnist_lines_dataset.py b/src/text_recognizer/datasets/emnist_lines_dataset.py
index d49319f..4d8b646 100644
--- a/src/text_recognizer/datasets/emnist_lines_dataset.py
+++ b/src/text_recognizer/datasets/emnist_lines_dataset.py
@@ -11,7 +11,8 @@ import torch
from torch.utils.data import Dataset
from torchvision.transforms import Compose, Normalize, ToTensor
-from text_recognizer.datasets import DATA_DIRNAME, EmnistDataset, SentenceGenerator
+from text_recognizer.datasets import DATA_DIRNAME, EmnistDataset
+from text_recognizer.datasets.sentence_generator import SentenceGenerator
from text_recognizer.datasets.util import Transpose
DATA_DIRNAME = DATA_DIRNAME / "processed" / "emnist_lines"