diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-06 23:23:02 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-06 23:23:02 +0100 |
| commit | 41b2ff656b927aee53549cba564dfd872ff4c9d2 (patch) | |
| tree | 8965e254cdabea09482fe81a0a2190ebbe1a8407 /tests/support/create_emnist_support_files.py | |
| parent | e3aad99e8bb672740bb47cba5a073f087a1eaa90 (diff) | |
chore: remove support files for old tests
Diffstat (limited to 'tests/support/create_emnist_support_files.py')
| -rw-r--r-- | tests/support/create_emnist_support_files.py | 30 |
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() |