summaryrefslogtreecommitdiff
path: root/text_recognizer/data/iam_synthetic_paragraphs.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-18 17:42:14 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-18 17:42:14 +0200
commit5fca1bb6aeb48f106750d66b46317acf38cd8a10 (patch)
tree326e916b7c054034a4558678fa54491bee45cc83 /text_recognizer/data/iam_synthetic_paragraphs.py
parentf132ea7e0299a4cdc57245d8339f10411c13bfbc (diff)
Add resize attribute to IAM paragraphs
Diffstat (limited to 'text_recognizer/data/iam_synthetic_paragraphs.py')
-rw-r--r--text_recognizer/data/iam_synthetic_paragraphs.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py
index 24ca896..b9cf90d 100644
--- a/text_recognizer/data/iam_synthetic_paragraphs.py
+++ b/text_recognizer/data/iam_synthetic_paragraphs.py
@@ -84,7 +84,7 @@ class IAMSyntheticParagraphs(IAMParagraphs):
data,
targets,
transform=get_transform(
- image_shape=self.dims[1:], augment=self.augment
+ image_shape=self.dims[1:], augment=self.augment, resize=self.resize
),
target_transform=get_target_transform(self.word_pieces),
)