diff options
Diffstat (limited to 'text_recognizer/data/emnist_lines.py')
| -rw-r--r-- | text_recognizer/data/emnist_lines.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py index 63c9f22..88aac0d 100644 --- a/text_recognizer/data/emnist_lines.py +++ b/text_recognizer/data/emnist_lines.py @@ -13,9 +13,9 @@ from text_recognizer.data.base_data_module import BaseDataModule, load_and_print from text_recognizer.data.base_dataset import BaseDataset, convert_strings_to_labels from text_recognizer.data.emnist import EMNIST from text_recognizer.data.mappings import EmnistMapping -from text_recognizer.data.transforms.load_transform import load_transform_from_file +from text_recognizer.data.stems.line import LineStem from text_recognizer.data.utils.sentence_generator import SentenceGenerator -from text_recognizer.metadata import emnist_lines as metadata +import text_recognizer.metadata.emnist_lines as metadata class EMNISTLines(BaseDataModule): @@ -250,6 +250,6 @@ def _create_dataset_of_images( def generate_emnist_lines() -> None: """Generates a synthetic handwritten dataset and displays info.""" - transform = load_transform_from_file("transform/emnist_lines.yaml") - test_transform = load_transform_from_file("test_transform/default.yaml") + transform = LineStem(augment=False) + test_transform = LineStem(augment=False) load_and_print_info(EMNISTLines(transform=transform, test_transform=test_transform)) |