summaryrefslogtreecommitdiff
path: root/text_recognizer/data
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-30 00:02:05 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-30 00:02:05 +0200
commit92fc1c7ed2f9f64552be8f71d9b8ab0d5a0a88d4 (patch)
tree68921ead6bffea3b7c93af365f1de58ecca130e0 /text_recognizer/data
parent34098ccbbbf6379c0bd29a987440b8479c743746 (diff)
Black formatting
Diffstat (limited to 'text_recognizer/data')
-rw-r--r--text_recognizer/data/iam_lines.py4
-rw-r--r--text_recognizer/data/iam_paragraphs.py4
2 files changed, 6 insertions, 2 deletions
diff --git a/text_recognizer/data/iam_lines.py b/text_recognizer/data/iam_lines.py
index 705cfa3..13dd379 100644
--- a/text_recognizer/data/iam_lines.py
+++ b/text_recognizer/data/iam_lines.py
@@ -43,7 +43,9 @@ class IAMLines(BaseDataModule):
augment: bool = attr.ib(default=True)
fraction: float = attr.ib(default=0.8)
- dims: Tuple[int, int, int] = attr.ib(init=False, default=(1, IMAGE_HEIGHT, IMAGE_WIDTH))
+ dims: Tuple[int, int, int] = attr.ib(
+ init=False, default=(1, IMAGE_HEIGHT, IMAGE_WIDTH)
+ )
output_dims: Tuple[int, int] = attr.ib(init=False, default=(MAX_LABEL_LENGTH, 1))
def __attrs_post_init__(self) -> None:
diff --git a/text_recognizer/data/iam_paragraphs.py b/text_recognizer/data/iam_paragraphs.py
index 9977978..de32875 100644
--- a/text_recognizer/data/iam_paragraphs.py
+++ b/text_recognizer/data/iam_paragraphs.py
@@ -41,7 +41,9 @@ class IAMParagraphs(BaseDataModule):
augment: bool = attr.ib(default=True)
train_fraction: float = attr.ib(default=0.8)
word_pieces: bool = attr.ib(default=False)
- dims: Tuple[int, int, int] = attr.ib(init=False, default=(1, IMAGE_HEIGHT, IMAGE_WIDTH))
+ dims: Tuple[int, int, int] = attr.ib(
+ init=False, default=(1, IMAGE_HEIGHT, IMAGE_WIDTH)
+ )
output_dims: Tuple[int, int] = attr.ib(init=False, default=(MAX_LABEL_LENGTH, 1))
def __attrs_post_init__(self) -> None: