diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-13 18:44:24 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-13 18:44:24 +0200 |
| commit | 191aa5a5c080ed7bf7f6d0408aa1cac4295c57d2 (patch) | |
| tree | 8b1472150d420486ae7493b9f7c95c0b14b6c459 /text_recognizer | |
| parent | bec6649f8202dd1a12209ed2e489ea4d22fc69e0 (diff) | |
Fix emnist datamodule
Diffstat (limited to 'text_recognizer')
| -rw-r--r-- | text_recognizer/data/emnist_lines.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py index 2107d74..b33c9e1 100644 --- a/text_recognizer/data/emnist_lines.py +++ b/text_recognizer/data/emnist_lines.py @@ -71,7 +71,7 @@ class EMNISTLines(BaseDataModule): self.num_val = num_val self.num_test = num_test - self.emnist = EMNIST(mapping=self.mapping) + self.emnist = EMNIST() max_width = ( int(self.emnist.dims[2] * (self.max_length + 1) * (1 - self.min_overlap)) + IMAGE_X_PADDING |