diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-26 23:17:14 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-26 23:17:14 +0100 |
| commit | 9b6c99172f80b45030279e7268aaa80e5a1ed574 (patch) | |
| tree | 5049b225927e786dd462265ec8bca9a0f8084f18 /text_recognizer/data | |
| parent | 5e81d11e18cc28c383a30c10c53b8b477ada3c0a (diff) | |
fix: lint and format
Diffstat (limited to 'text_recognizer/data')
| -rw-r--r-- | text_recognizer/data/iam_paragraphs.py | 5 | ||||
| -rw-r--r-- | text_recognizer/data/iam_synthetic_paragraphs.py | 8 |
2 files changed, 2 insertions, 11 deletions
diff --git a/text_recognizer/data/iam_paragraphs.py b/text_recognizer/data/iam_paragraphs.py index a8d5e0d..033b93e 100644 --- a/text_recognizer/data/iam_paragraphs.py +++ b/text_recognizer/data/iam_paragraphs.py @@ -86,10 +86,7 @@ class IAMParagraphs(BaseDataModule): length=self.output_dims[0], ) return BaseDataset( - data, - targets, - transform=transform, - target_transform=target_transform, + data, targets, transform=transform, target_transform=target_transform, ) log.info(f"Loading IAM paragraph regions and lines for {stage}...") diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py index 351e052..ea59098 100644 --- a/text_recognizer/data/iam_synthetic_paragraphs.py +++ b/text_recognizer/data/iam_synthetic_paragraphs.py @@ -46,13 +46,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( - 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") |