blob: f9ff0304b6a778579bf4745b87a7a8b91e24f198 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
"""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()
|
