From 90090f4b66284ce616ea475dfb04e2ce0e4fdf64 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 8 Jun 2022 08:39:32 +0200 Subject: WIP mappings --- text_recognizer/data/emnist.py | 16 ---------------- 1 file changed, 16 deletions(-) (limited to 'text_recognizer/data/emnist.py') diff --git a/text_recognizer/data/emnist.py b/text_recognizer/data/emnist.py index 1b1381a..ea27984 100644 --- a/text_recognizer/data/emnist.py +++ b/text_recognizer/data/emnist.py @@ -102,22 +102,6 @@ class EMNIST(BaseDataModule): return basic + data -def emnist_mapping( - extra_symbols: Optional[Set[str]] = None, -) -> Tuple[List, Dict[str, int], List[int]]: - """Return the EMNIST mapping.""" - if not ESSENTIALS_FILENAME.exists(): - download_and_process_emnist() - with ESSENTIALS_FILENAME.open() as f: - essentials = json.load(f) - mapping = list(essentials["characters"]) - if extra_symbols is not None: - mapping += extra_symbols - inverse_mapping = {v: k for k, v in enumerate(mapping)} - input_shape = essentials["input_shape"] - return mapping, inverse_mapping, input_shape - - def download_and_process_emnist() -> None: """Downloads and preprocesses EMNIST dataset.""" metadata = toml.load(METADATA_FILENAME) -- cgit v1.2.3-70-g09d2