From 9b8e14d89f0ef2508ed11f994f73af624155fe1d Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 27 Sep 2022 01:44:49 +0200 Subject: Update data modules --- text_recognizer/data/emnist_lines.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'text_recognizer/data/emnist_lines.py') 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)) -- cgit v1.2.3-70-g09d2