diff options
Diffstat (limited to 'text_recognizer/data/iam_synthetic_paragraphs.py')
| -rw-r--r-- | text_recognizer/data/iam_synthetic_paragraphs.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py index d906399..351e052 100644 --- a/text_recognizer/data/iam_synthetic_paragraphs.py +++ b/text_recognizer/data/iam_synthetic_paragraphs.py @@ -46,7 +46,13 @@ class IAMSyntheticParagraphs(IAMParagraphs): log.info("Preparing IAM lines for synthetic paragraphs dataset.") log.info("Cropping IAM line regions and loading labels.") - iam = IAM(mapping=EmnistMapping(extra_symbols={NEW_LINE_TOKEN,})) + iam = IAM( + mapping=EmnistMapping( + extra_symbols={ + NEW_LINE_TOKEN, + } + ) + ) iam.prepare_data() crops_train, labels_train = line_crops_and_labels(iam, "train") |