diff options
Diffstat (limited to 'text_recognizer/data/mappings')
| -rw-r--r-- | text_recognizer/data/mappings/base.py (renamed from text_recognizer/data/mappings/base_mapping.py) | 0 | ||||
| -rw-r--r-- | text_recognizer/data/mappings/emnist.py (renamed from text_recognizer/data/mappings/emnist_mapping.py) | 2 | ||||
| -rw-r--r-- | text_recognizer/data/mappings/word_piece.py (renamed from text_recognizer/data/mappings/word_piece_mapping.py) | 6 |
3 files changed, 3 insertions, 5 deletions
diff --git a/text_recognizer/data/mappings/base_mapping.py b/text_recognizer/data/mappings/base.py index 572ac95..572ac95 100644 --- a/text_recognizer/data/mappings/base_mapping.py +++ b/text_recognizer/data/mappings/base.py diff --git a/text_recognizer/data/mappings/emnist_mapping.py b/text_recognizer/data/mappings/emnist.py index 3eed3d8..655169e 100644 --- a/text_recognizer/data/mappings/emnist_mapping.py +++ b/text_recognizer/data/mappings/emnist.py @@ -4,8 +4,8 @@ from typing import List, Optional, Set, Union import torch from torch import Tensor -from text_recognizer.data.mappings.base_mapping import AbstractMapping from text_recognizer.data.emnist import emnist_mapping +from text_recognizer.data.mappings.base import AbstractMapping class EmnistMapping(AbstractMapping): diff --git a/text_recognizer/data/mappings/word_piece_mapping.py b/text_recognizer/data/mappings/word_piece.py index f9e4e7a..861c3bd 100644 --- a/text_recognizer/data/mappings/word_piece_mapping.py +++ b/text_recognizer/data/mappings/word_piece.py @@ -1,12 +1,10 @@ """Word piece mapping.""" -from pathlib import Path -from typing import List, Optional, Set, Union +from typing import List, Set, Union -from loguru import logger as log import torch from torch import Tensor -from text_recognizer.data.mappings.emnist_mapping import EmnistMapping +from text_recognizer.data.mappings.emnist import EmnistMapping from text_recognizer.data.utils.iam_preprocessor import Preprocessor |