diff options
| -rw-r--r-- | text_recognizer/data/stems/__init__.py | 0 | ||||
| -rw-r--r-- | text_recognizer/data/transforms/image.py (renamed from text_recognizer/data/stems/image.py) | 0 | ||||
| -rw-r--r-- | text_recognizer/data/transforms/line.py (renamed from text_recognizer/data/stems/line.py) | 0 | ||||
| -rw-r--r-- | text_recognizer/data/transforms/paragraph.py (renamed from text_recognizer/data/stems/paragraph.py) | 0 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 4 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 4 | ||||
| -rw-r--r-- | training/conf/experiment/conv_transformer_lines.yaml | 3 |
7 files changed, 4 insertions, 7 deletions
diff --git a/text_recognizer/data/stems/__init__.py b/text_recognizer/data/stems/__init__.py deleted file mode 100644 index e69de29..0000000 --- a/text_recognizer/data/stems/__init__.py +++ /dev/null diff --git a/text_recognizer/data/stems/image.py b/text_recognizer/data/transforms/image.py index f04b3a0..f04b3a0 100644 --- a/text_recognizer/data/stems/image.py +++ b/text_recognizer/data/transforms/image.py diff --git a/text_recognizer/data/stems/line.py b/text_recognizer/data/transforms/line.py index 4f0ce05..4f0ce05 100644 --- a/text_recognizer/data/stems/line.py +++ b/text_recognizer/data/transforms/line.py diff --git a/text_recognizer/data/stems/paragraph.py b/text_recognizer/data/transforms/paragraph.py index 39e1e59..39e1e59 100644 --- a/text_recognizer/data/stems/paragraph.py +++ b/text_recognizer/data/transforms/paragraph.py diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index 64c3964..e4ef896 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -4,10 +4,10 @@ num_workers: 12 train_fraction: 0.8 pin_memory: true transform: - _target_: text_recognizer.data.stems.paragraph.ParagraphStem + _target_: text_recognizer.data.transforms.paragraph.ParagraphStem augment: true test_transform: - _target_: text_recognizer.data.stems.paragraph.ParagraphStem + _target_: text_recognizer.data.transforms.paragraph.ParagraphStem augment: false target_transform: _target_: text_recognizer.data.transforms.pad.Pad diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index f84116d..1205c75 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -4,10 +4,10 @@ num_workers: 12 train_fraction: 0.9 pin_memory: true transform: - _target_: text_recognizer.data.stems.line.IamLinesStem + _target_: text_recognizer.data.transforms.line.IamLinesStem augment: true test_transform: - _target_: text_recognizer.data.stems.line.IamLinesStem + _target_: text_recognizer.data.transforms.line.IamLinesStem augment: false tokenizer: _target_: text_recognizer.data.tokenizer.Tokenizer diff --git a/training/conf/experiment/conv_transformer_lines.yaml b/training/conf/experiment/conv_transformer_lines.yaml index 3f5da86..948968a 100644 --- a/training/conf/experiment/conv_transformer_lines.yaml +++ b/training/conf/experiment/conv_transformer_lines.yaml @@ -54,9 +54,6 @@ lr_scheduler: datamodule: batch_size: 16 train_fraction: 0.95 - transform: - _target_: text_recognizer.data.stems.line.IamLinesStem - augment: false network: _target_: text_recognizer.networks.ConvTransformer |