summaryrefslogtreecommitdiff
path: root/text_recognizer/data/emnist_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/emnist_lines.py
parent2cc6aa059139b57057609817913ad515063c2eab (diff)
Formats
Diffstat (limited to 'text_recognizer/data/emnist_lines.py')
-rw-r--r--text_recognizer/data/emnist_lines.py10
1 files changed, 3 insertions, 7 deletions
diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py
index b33c9e1..c36132e 100644
--- a/text_recognizer/data/emnist_lines.py
+++ b/text_recognizer/data/emnist_lines.py
@@ -4,23 +4,19 @@ from pathlib import Path
from typing import Callable, DefaultDict, List, Optional, Tuple, Type
import h5py
-from loguru import logger as log
import numpy as np
import torch
-from torch import Tensor
import torchvision.transforms as T
+from loguru import logger as log
+from torch import Tensor
-from text_recognizer.data.base_data_module import (
- BaseDataModule,
- load_and_print_info,
-)
+from text_recognizer.data.base_data_module import BaseDataModule, load_and_print_info
from text_recognizer.data.base_dataset import BaseDataset, convert_strings_to_labels
from text_recognizer.data.emnist import EMNIST
from text_recognizer.data.mappings import EmnistMapping
from text_recognizer.data.transforms.load_transform import load_transform_from_file
from text_recognizer.data.utils.sentence_generator import SentenceGenerator
-
DATA_DIRNAME = BaseDataModule.data_dirname() / "processed" / "emnist_lines"
ESSENTIALS_FILENAME = (
Path(__file__).parents[0].resolve() / "mappings" / "emnist_lines_essentials.json"