diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:11:18 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:11:18 +0200 |
| commit | 2e7d569cede02667f1eabe4d32a2ed97f565c515 (patch) | |
| tree | 674e76871de64649204781cf6e3a564c4bd09c9c /text_recognizer/data | |
| parent | b291781c3f4599e7320388ee6c26178e4e371dae (diff) | |
Rearrange import order and formatting
Diffstat (limited to 'text_recognizer/data')
| -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 |