diff options
| -rw-r--r-- | text_recognizer/data/emnist.py | 2 | ||||
| -rw-r--r-- | text_recognizer/data/emnist_lines.py | 2 | ||||
| -rw-r--r-- | text_recognizer/data/iam_extended_paragraphs.py | 1 | ||||
| -rw-r--r-- | text_recognizer/data/iam_lines.py | 2 |
4 files changed, 3 insertions, 4 deletions
diff --git a/text_recognizer/data/emnist.py b/text_recognizer/data/emnist.py index e2bc5b9..dc8d31a 100644 --- a/text_recognizer/data/emnist.py +++ b/text_recognizer/data/emnist.py @@ -17,8 +17,8 @@ from text_recognizer.data.base_data_module import ( load_and_print_info, ) from text_recognizer.data.base_dataset import BaseDataset, split_dataset -from text_recognizer.data.utils.download_utils import download_dataset from text_recognizer.data.transforms.load_transform import load_transform_from_file +from text_recognizer.data.utils.download_utils import download_dataset SEED = 4711 diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py index 1a64931..c267286 100644 --- a/text_recognizer/data/emnist_lines.py +++ b/text_recognizer/data/emnist_lines.py @@ -17,8 +17,8 @@ from text_recognizer.data.base_data_module import ( ) from text_recognizer.data.base_dataset import BaseDataset, convert_strings_to_labels from text_recognizer.data.emnist import EMNIST -from text_recognizer.data.utils.sentence_generator import SentenceGenerator from text_recognizer.data.transforms.load_transform import load_transform_from_file +from text_recognizer.data.utils.sentence_generator import SentenceGenerator DATA_DIRNAME = BaseDataModule.data_dirname() / "processed" / "emnist_lines" diff --git a/text_recognizer/data/iam_extended_paragraphs.py b/text_recognizer/data/iam_extended_paragraphs.py index 87b8ef1..f43e3f7 100644 --- a/text_recognizer/data/iam_extended_paragraphs.py +++ b/text_recognizer/data/iam_extended_paragraphs.py @@ -1,6 +1,5 @@ """IAM original and sythetic dataset class.""" import attr - from torch.utils.data import ConcatDataset from text_recognizer.data.base_data_module import BaseDataModule, load_and_print_info diff --git a/text_recognizer/data/iam_lines.py b/text_recognizer/data/iam_lines.py index 672a6f0..aa5473a 100644 --- a/text_recognizer/data/iam_lines.py +++ b/text_recognizer/data/iam_lines.py @@ -20,9 +20,9 @@ from text_recognizer.data.base_dataset import ( split_dataset, ) from text_recognizer.data.iam import IAM -from text_recognizer.data.utils import image_utils from text_recognizer.data.mappings.emnist import EmnistMapping from text_recognizer.data.transforms.load_transform import load_transform_from_file +from text_recognizer.data.utils import image_utils ImageFile.LOAD_TRUNCATED_IMAGES = True |