diff options
Diffstat (limited to 'text_recognizer/data')
| -rw-r--r-- | text_recognizer/data/base_data_module.py | 1 | ||||
| -rw-r--r-- | text_recognizer/data/base_dataset.py | 2 | ||||
| -rw-r--r-- | text_recognizer/data/transforms/pad.py | 1 |
3 files changed, 1 insertions, 3 deletions
diff --git a/text_recognizer/data/base_data_module.py b/text_recognizer/data/base_data_module.py index 3cb0aa2..6efa4fb 100644 --- a/text_recognizer/data/base_data_module.py +++ b/text_recognizer/data/base_data_module.py @@ -1,5 +1,4 @@ """Base lightning DataModule class.""" -from pathlib import Path from typing import Callable, Dict, Optional, Tuple, TypeVar import pytorch_lightning as L diff --git a/text_recognizer/data/base_dataset.py b/text_recognizer/data/base_dataset.py index b840bc8..12f524c 100644 --- a/text_recognizer/data/base_dataset.py +++ b/text_recognizer/data/base_dataset.py @@ -1,5 +1,5 @@ """Base PyTorch Dataset class.""" -from typing import Callable, Dict, Optional, Sequence, Tuple, Union +from typing import Callable, Dict, Sequence, Tuple, Union import torch from torch import Tensor diff --git a/text_recognizer/data/transforms/pad.py b/text_recognizer/data/transforms/pad.py index df1d83f..baf637a 100644 --- a/text_recognizer/data/transforms/pad.py +++ b/text_recognizer/data/transforms/pad.py @@ -1,7 +1,6 @@ """Pad targets to equal length.""" import torch -import torch.functional as F from torch import Tensor |