From 9ce21f569ecac03f15f2ad669fde3dd4a512f8cc Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 15 Apr 2024 21:47:54 +0200 Subject: Format --- text_recognizer/data/transforms/image.py | 4 ++-- text_recognizer/data/transforms/line.py | 2 +- text_recognizer/data/transforms/paragraph.py | 1 - 3 files changed, 3 insertions(+), 4 deletions(-) (limited to 'text_recognizer/data/transforms') diff --git a/text_recognizer/data/transforms/image.py b/text_recognizer/data/transforms/image.py index f04b3a0..05c9d94 100644 --- a/text_recognizer/data/transforms/image.py +++ b/text_recognizer/data/transforms/image.py @@ -1,7 +1,7 @@ -from PIL import Image import torch -from torch import Tensor import torchvision.transforms as T +from PIL import Image +from torch import Tensor class ImageStem: diff --git a/text_recognizer/data/transforms/line.py b/text_recognizer/data/transforms/line.py index 6c38213..e4473eb 100644 --- a/text_recognizer/data/transforms/line.py +++ b/text_recognizer/data/transforms/line.py @@ -1,8 +1,8 @@ import random from typing import Any, Dict -from PIL import Image import torchvision.transforms as T +from PIL import Image import text_recognizer.metadata.iam_lines as metadata from text_recognizer.data.transforms.image import ImageStem diff --git a/text_recognizer/data/transforms/paragraph.py b/text_recognizer/data/transforms/paragraph.py index b364f91..639bb59 100644 --- a/text_recognizer/data/transforms/paragraph.py +++ b/text_recognizer/data/transforms/paragraph.py @@ -4,7 +4,6 @@ import torchvision.transforms as T import text_recognizer.metadata.iam_paragraphs as metadata from text_recognizer.data.transforms.image import ImageStem - IMAGE_HEIGHT, IMAGE_WIDTH = metadata.IMAGE_HEIGHT, metadata.IMAGE_WIDTH IMAGE_SHAPE = metadata.IMAGE_SHAPE -- cgit v1.2.3-70-g09d2