summaryrefslogtreecommitdiff
path: root/text_recognizer/data/iam_synthetic_paragraphs.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/data/iam_synthetic_paragraphs.py')
-rw-r--r--text_recognizer/data/iam_synthetic_paragraphs.py12
1 files changed, 4 insertions, 8 deletions
diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py
index d51e010..5e66499 100644
--- a/text_recognizer/data/iam_synthetic_paragraphs.py
+++ b/text_recognizer/data/iam_synthetic_paragraphs.py
@@ -2,15 +2,12 @@
import random
from typing import Any, Callable, List, Optional, Sequence, Tuple
-from loguru import logger as log
import numpy as np
+from loguru import logger as log
from PIL import Image
from text_recognizer.data.base_data_module import BaseDataModule, load_and_print_info
-from text_recognizer.data.base_dataset import (
- BaseDataset,
- convert_strings_to_labels,
-)
+from text_recognizer.data.base_dataset import BaseDataset, convert_strings_to_labels
from text_recognizer.data.iam import IAM
from text_recognizer.data.iam_lines import (
line_crops_and_labels,
@@ -18,16 +15,15 @@ from text_recognizer.data.iam_lines import (
save_images_and_labels,
)
from text_recognizer.data.iam_paragraphs import (
- get_dataset_properties,
- IAMParagraphs,
IMAGE_SCALE_FACTOR,
NEW_LINE_TOKEN,
+ IAMParagraphs,
+ get_dataset_properties,
resize_image,
)
from text_recognizer.data.mappings import EmnistMapping
from text_recognizer.data.transforms.load_transform import load_transform_from_file
-
PROCESSED_DATA_DIRNAME = (
BaseDataModule.data_dirname() / "processed" / "iam_synthetic_paragraphs"
)