summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/data/iam_lines.py2
-rw-r--r--text_recognizer/data/iam_paragraphs.py2
-rw-r--r--text_recognizer/data/iam_synthetic_paragraphs.py2
-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
6 files changed, 6 insertions, 8 deletions
diff --git a/text_recognizer/data/iam_lines.py b/text_recognizer/data/iam_lines.py
index efd1cde..672a6f0 100644
--- a/text_recognizer/data/iam_lines.py
+++ b/text_recognizer/data/iam_lines.py
@@ -20,8 +20,8 @@ from text_recognizer.data.base_dataset import (
split_dataset,
)
from text_recognizer.data.iam import IAM
-from text_recognizer.data.mappings.emnist_mapping import EmnistMapping
from text_recognizer.data.utils import image_utils
+from text_recognizer.data.mappings.emnist import EmnistMapping
from text_recognizer.data.transforms.load_transform import load_transform_from_file
diff --git a/text_recognizer/data/iam_paragraphs.py b/text_recognizer/data/iam_paragraphs.py
index 26674e0..033b93e 100644
--- a/text_recognizer/data/iam_paragraphs.py
+++ b/text_recognizer/data/iam_paragraphs.py
@@ -17,7 +17,7 @@ from text_recognizer.data.base_dataset import (
split_dataset,
)
from text_recognizer.data.iam import IAM
-from text_recognizer.data.mappings.emnist_mapping import EmnistMapping
+from text_recognizer.data.mappings.emnist import EmnistMapping
from text_recognizer.data.transforms.load_transform import load_transform_from_file
diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py
index d6fea43..d906399 100644
--- a/text_recognizer/data/iam_synthetic_paragraphs.py
+++ b/text_recognizer/data/iam_synthetic_paragraphs.py
@@ -25,7 +25,7 @@ from text_recognizer.data.iam_paragraphs import (
NEW_LINE_TOKEN,
resize_image,
)
-from text_recognizer.data.mappings.emnist_mapping import EmnistMapping
+from text_recognizer.data.mappings.emnist import EmnistMapping
from text_recognizer.data.transforms.load_transform import load_transform_from_file
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