summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-11-15 16:24:12 +0100
committeraktersnurra <gustaf.rydholm@gmail.com>2020-11-15 16:24:12 +0100
commitcad676fc423efeafde65f03e4815248f2d357011 (patch)
tree43ac8547abf1bfd7bb67ebe60e937c34765197b2
parentf9a222dc1dd351147ed9aec970097b9f97995403 (diff)
Small changes.
-rw-r--r--src/text_recognizer/tests/support/create_emnist_lines_support_files.py5
-rw-r--r--src/text_recognizer/tests/support/create_iam_lines_support_files.py5
2 files changed, 8 insertions, 2 deletions
diff --git a/src/text_recognizer/tests/support/create_emnist_lines_support_files.py b/src/text_recognizer/tests/support/create_emnist_lines_support_files.py
index b200ff5..4496e40 100644
--- a/src/text_recognizer/tests/support/create_emnist_lines_support_files.py
+++ b/src/text_recognizer/tests/support/create_emnist_lines_support_files.py
@@ -2,6 +2,8 @@
from pathlib import Path
import shutil
+import numpy as np
+
from text_recognizer.datasets import EmnistLinesDataset
import text_recognizer.util as util
@@ -10,6 +12,7 @@ SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "emnist_lines"
def create_emnist_lines_support_files() -> None:
+ """Create EMNIST Lines test images."""
shutil.rmtree(SUPPORT_DIRNAME, ignore_errors=True)
SUPPORT_DIRNAME.mkdir()
@@ -27,7 +30,7 @@ def create_emnist_lines_support_files() -> None:
for label in np.argmax(target[1:], dim=-1).flatten()
)
.stip()
- .strip(self.mapper.pad_token)
+ .strip(dataset.mapper.pad_token)
)
print(label)
diff --git a/src/text_recognizer/tests/support/create_iam_lines_support_files.py b/src/text_recognizer/tests/support/create_iam_lines_support_files.py
index 15d0e4e..bb568ee 100644
--- a/src/text_recognizer/tests/support/create_iam_lines_support_files.py
+++ b/src/text_recognizer/tests/support/create_iam_lines_support_files.py
@@ -2,6 +2,8 @@
from pathlib import Path
import shutil
+import numpy as np
+
from text_recognizer.datasets import IamLinesDataset
import text_recognizer.util as util
@@ -10,6 +12,7 @@ SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "iam_lines"
def create_emnist_lines_support_files() -> None:
+ """Create IAM Lines test images."""
shutil.rmtree(SUPPORT_DIRNAME, ignore_errors=True)
SUPPORT_DIRNAME.mkdir()
@@ -27,7 +30,7 @@ def create_emnist_lines_support_files() -> None:
for label in np.argmax(target[1:], dim=-1).flatten()
)
.stip()
- .strip(self.mapper.pad_token)
+ .strip(dataset.mapper.pad_token)
)
print(label)