diff options
Diffstat (limited to 'text_recognizer/data/transforms')
| -rw-r--r-- | text_recognizer/data/transforms/line.py | 2 | ||||
| -rw-r--r-- | text_recognizer/data/transforms/paragraph.py | 2 |
2 files changed, 2 insertions, 2 deletions
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 |