diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:10:14 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-27 22:10:14 +0200 |
| commit | 07a5b84b001756b06c6dc0c54f396ea42b11067b (patch) | |
| tree | d878d498d7bd4aebab1194e6d9851dc16e0b6b8e /text_recognizer/data/mappings | |
| parent | eb672fa209d4a34124cce06bd8737164647a2b1b (diff) | |
Rename mappings
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 |