diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-03 18:18:48 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-03 18:18:48 +0200 |
| commit | bd4bd443f339e95007bfdabf3e060db720f4d4b9 (patch) | |
| tree | e55cb3744904f7c2a0348b100c7e92a65e538a16 /text_recognizer/data/iam_synthetic_paragraphs.py | |
| parent | 75801019981492eedf9280cb352eea3d8e99b65f (diff) | |
Training working, multiple bug fixes
Diffstat (limited to 'text_recognizer/data/iam_synthetic_paragraphs.py')
| -rw-r--r-- | text_recognizer/data/iam_synthetic_paragraphs.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py index c938f8b..24ca896 100644 --- a/text_recognizer/data/iam_synthetic_paragraphs.py +++ b/text_recognizer/data/iam_synthetic_paragraphs.py @@ -21,7 +21,7 @@ from text_recognizer.data.iam_paragraphs import ( IMAGE_SCALE_FACTOR, resize_image, ) -from text_recognizer.data.mappings import EmnistMapping +from text_recognizer.data.emnist_mapping import EmnistMapping from text_recognizer.data.iam import IAM from text_recognizer.data.iam_lines import ( line_crops_and_labels, @@ -47,7 +47,7 @@ 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()) + iam = IAM(mapping=EmnistMapping(extra_symbols={NEW_LINE_TOKEN,})) iam.prepare_data() crops_train, labels_train = line_crops_and_labels(iam, "train") |