diff options
Diffstat (limited to 'text_recognizer/data')
-rw-r--r-- | text_recognizer/data/emnist_lines.py | 2 | ||||
-rw-r--r-- | text_recognizer/data/iam_extended_paragraphs.py | 2 | ||||
-rw-r--r-- | text_recognizer/data/iam_lines.py | 2 | ||||
-rw-r--r-- | text_recognizer/data/iam_paragraphs.py | 2 | ||||
-rw-r--r-- | text_recognizer/data/iam_synthetic_paragraphs.py | 2 | ||||
-rw-r--r-- | text_recognizer/data/transforms/line.py | 2 | ||||
-rw-r--r-- | text_recognizer/data/transforms/paragraph.py | 2 |
7 files changed, 7 insertions, 7 deletions
diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py index 8a31c44..4ae4787 100644 --- a/text_recognizer/data/emnist_lines.py +++ b/text_recognizer/data/emnist_lines.py @@ -13,7 +13,7 @@ from text_recognizer.data.base_data_module import BaseDataModule, load_and_print from text_recognizer.data.base_dataset import BaseDataset, convert_strings_to_labels from text_recognizer.data.emnist import EMNIST from text_recognizer.data.tokenizer import Tokenizer -from text_recognizer.data.stems.line import LineStem +from text_recognizer.data.transforms.line import LineStem from text_recognizer.data.utils.sentence_generator import SentenceGenerator import text_recognizer.metadata.emnist_lines as metadata diff --git a/text_recognizer/data/iam_extended_paragraphs.py b/text_recognizer/data/iam_extended_paragraphs.py index c6628a8..3f3500f 100644 --- a/text_recognizer/data/iam_extended_paragraphs.py +++ b/text_recognizer/data/iam_extended_paragraphs.py @@ -8,7 +8,7 @@ from text_recognizer.data.iam_paragraphs import IAMParagraphs from text_recognizer.data.iam_synthetic_paragraphs import IAMSyntheticParagraphs from text_recognizer.data.transforms.pad import Pad from text_recognizer.data.tokenizer import Tokenizer -from text_recognizer.data.stems.paragraph import ParagraphStem +from text_recognizer.data.transforms.paragraph import ParagraphStem import text_recognizer.metadata.iam_paragraphs as metadata diff --git a/text_recognizer/data/iam_lines.py b/text_recognizer/data/iam_lines.py index a0d9b59..9ab3624 100644 --- a/text_recognizer/data/iam_lines.py +++ b/text_recognizer/data/iam_lines.py @@ -20,7 +20,7 @@ from text_recognizer.data.base_dataset import ( ) from text_recognizer.data.iam import IAM from text_recognizer.data.tokenizer import Tokenizer -from text_recognizer.data.stems.line import IamLinesStem +from text_recognizer.data.transforms.line import IamLinesStem from text_recognizer.data.utils import image_utils import text_recognizer.metadata.iam_lines as metadata diff --git a/text_recognizer/data/iam_paragraphs.py b/text_recognizer/data/iam_paragraphs.py index a078c7d..0d53d6b 100644 --- a/text_recognizer/data/iam_paragraphs.py +++ b/text_recognizer/data/iam_paragraphs.py @@ -18,7 +18,7 @@ from text_recognizer.data.base_dataset import ( from text_recognizer.data.iam import IAM from text_recognizer.data.transforms.pad import Pad from text_recognizer.data.tokenizer import Tokenizer -from text_recognizer.data.stems.paragraph import ParagraphStem +from text_recognizer.data.transforms.paragraph import ParagraphStem import text_recognizer.metadata.iam_paragraphs as metadata diff --git a/text_recognizer/data/iam_synthetic_paragraphs.py b/text_recognizer/data/iam_synthetic_paragraphs.py index 511a8d4..5743bcc 100644 --- a/text_recognizer/data/iam_synthetic_paragraphs.py +++ b/text_recognizer/data/iam_synthetic_paragraphs.py @@ -20,7 +20,7 @@ from text_recognizer.data.iam_lines import ( save_images_and_labels, ) from text_recognizer.data.tokenizer import Tokenizer -from text_recognizer.data.stems.paragraph import ParagraphStem +from text_recognizer.data.transforms.paragraph import ParagraphStem from text_recognizer.data.transforms.pad import Pad import text_recognizer.metadata.iam_synthetic_paragraphs as metadata diff --git a/text_recognizer/data/transforms/line.py b/text_recognizer/data/transforms/line.py index 4f0ce05..6c38213 100644 --- a/text_recognizer/data/transforms/line.py +++ b/text_recognizer/data/transforms/line.py @@ -5,7 +5,7 @@ from PIL import Image import torchvision.transforms as T import text_recognizer.metadata.iam_lines as metadata -from text_recognizer.data.stems.image import ImageStem +from text_recognizer.data.transforms.image import ImageStem class LineStem(ImageStem): diff --git a/text_recognizer/data/transforms/paragraph.py b/text_recognizer/data/transforms/paragraph.py index 39e1e59..b364f91 100644 --- a/text_recognizer/data/transforms/paragraph.py +++ b/text_recognizer/data/transforms/paragraph.py @@ -2,7 +2,7 @@ import torchvision.transforms as T import text_recognizer.metadata.iam_paragraphs as metadata -from text_recognizer.data.stems.image import ImageStem +from text_recognizer.data.transforms.image import ImageStem IMAGE_HEIGHT, IMAGE_WIDTH = metadata.IMAGE_HEIGHT, metadata.IMAGE_WIDTH |