summaryrefslogtreecommitdiff
path: root/tests/support/create_emnist_support_files.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/support/create_emnist_support_files.py')
-rw-r--r--tests/support/create_emnist_support_files.py30
1 files changed, 0 insertions, 30 deletions
diff --git a/tests/support/create_emnist_support_files.py b/tests/support/create_emnist_support_files.py
deleted file mode 100644
index f9ff030..0000000
--- a/tests/support/create_emnist_support_files.py
+++ /dev/null
@@ -1,30 +0,0 @@
-"""Module for creating EMNIST test support files."""
-from pathlib import Path
-import shutil
-
-from text_recognizer.datasets import EmnistDataset
-from text_recognizer.util import write_image
-
-SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "emnist"
-
-
-def create_emnist_support_files() -> None:
- """Create support images for test of CharacterPredictor class."""
- shutil.rmtree(SUPPORT_DIRNAME, ignore_errors=True)
- SUPPORT_DIRNAME.mkdir()
-
- dataset = EmnistDataset(train=False)
- dataset.load_or_generate_data()
-
- for index in [5, 7, 9]:
- image, label = dataset[index]
- if len(image.shape) == 3:
- image = image.squeeze(0)
- image = image.numpy()
- label = dataset.mapper(int(label))
- print(index, label)
- write_image(image, str(SUPPORT_DIRNAME / f"{label}.png"))
-
-
-if __name__ == "__main__":
- create_emnist_support_files()