summaryrefslogtreecommitdiff
path: root/text_recognizer/data/iam_lines.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-18 18:12:54 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-18 18:12:54 +0200
commit8eec5f3d4ce41c2ba30f0f1ab3f506fc3c0dd263 (patch)
tree19244a467e90ed8b919cdf2912d9e1b141ea1801 /text_recognizer/data/iam_lines.py
parent2cc6aa059139b57057609817913ad515063c2eab (diff)
Formats
Diffstat (limited to 'text_recognizer/data/iam_lines.py')
-rw-r--r--text_recognizer/data/iam_lines.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/text_recognizer/data/iam_lines.py b/text_recognizer/data/iam_lines.py
index 4899a48..a55ff1c 100644
--- a/text_recognizer/data/iam_lines.py
+++ b/text_recognizer/data/iam_lines.py
@@ -7,8 +7,8 @@ import json
from pathlib import Path
from typing import Callable, List, Optional, Sequence, Tuple, Type
-from loguru import logger as log
import numpy as np
+from loguru import logger as log
from PIL import Image, ImageFile, ImageOps
from torch import Tensor
@@ -23,7 +23,6 @@ from text_recognizer.data.mappings import EmnistMapping
from text_recognizer.data.transforms.load_transform import load_transform_from_file
from text_recognizer.data.utils import image_utils
-
ImageFile.LOAD_TRUNCATED_IMAGES = True
SEED = 4711