summaryrefslogtreecommitdiff
path: root/text_recognizer/data/transforms
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-10-04 22:06:07 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-10-04 22:06:07 +0200
commit8626406b872f7cf4b69a9dcc8bdfb777c6f1300d (patch)
tree7d5e6ce7acc968efee40e1945bf98a8fe5583481 /text_recognizer/data/transforms
parenta7393994256fea7ae57daa22a4a7b776ccf5cede (diff)
Update stem import paths
Diffstat (limited to 'text_recognizer/data/transforms')
-rw-r--r--text_recognizer/data/transforms/line.py2
-rw-r--r--text_recognizer/data/transforms/paragraph.py2
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