summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/transform
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-27 01:44:49 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-27 01:44:49 +0200
commit9b8e14d89f0ef2508ed11f994f73af624155fe1d (patch)
tree10d9c138f4449646c8b5c0f95003ba86b831d04d /training/conf/datamodule/transform
parent63376b1c2da81c23ad5239f908b640cd42a514c7 (diff)
Update data modules
Diffstat (limited to 'training/conf/datamodule/transform')
-rw-r--r--training/conf/datamodule/transform/default.yaml2
-rw-r--r--training/conf/datamodule/transform/emnist_lines.yaml14
-rw-r--r--training/conf/datamodule/transform/lines.yaml23
-rw-r--r--training/conf/datamodule/transform/paragraphs.yaml30
4 files changed, 0 insertions, 69 deletions
diff --git a/training/conf/datamodule/transform/default.yaml b/training/conf/datamodule/transform/default.yaml
deleted file mode 100644
index d6f6d78..0000000
--- a/training/conf/datamodule/transform/default.yaml
+++ /dev/null
@@ -1,2 +0,0 @@
-to_tensor:
- _target_: torchvision.transforms.ToTensor
diff --git a/training/conf/datamodule/transform/emnist_lines.yaml b/training/conf/datamodule/transform/emnist_lines.yaml
deleted file mode 100644
index 986ba20..0000000
--- a/training/conf/datamodule/transform/emnist_lines.yaml
+++ /dev/null
@@ -1,14 +0,0 @@
-to_tensor:
- _target_: torchvision.transforms.ToTensor
-
-color_jitter:
- _target_: torchvision.transforms.ColorJitter
- brightness: [0.5, 1.0]
-
-random_affine:
- _target_: torchvision.transforms.RandomAffine
- degrees: 3
- translate: [0.0, 0.05]
- scale: [0.4, 1.1]
- shear: [-40, 50]
- fill: 0
diff --git a/training/conf/datamodule/transform/lines.yaml b/training/conf/datamodule/transform/lines.yaml
deleted file mode 100644
index 6bd35ef..0000000
--- a/training/conf/datamodule/transform/lines.yaml
+++ /dev/null
@@ -1,23 +0,0 @@
-embed_crop:
- _target_: text_recognizer.data.transforms.embed_crop.EmbedCrop
- augment: false
-
-color_jitter:
- _target_: torchvision.transforms.ColorJitter
- brightness: [0.8, 1.6]
-
-random_affine:
- _target_: torchvision.transforms.RandomAffine
- degrees: 1
- shear: [-10, 10]
- interpolation: BILINEAR
- fill: 0
-
-random_perspective:
- _target_: torchvision.transforms.RandomPerspective
- distortion_scale: 0.05
- p: 0.25
- fill: 0
-
-to_tensor:
- _target_: torchvision.transforms.ToTensor
diff --git a/training/conf/datamodule/transform/paragraphs.yaml b/training/conf/datamodule/transform/paragraphs.yaml
deleted file mode 100644
index 4f260d0..0000000
--- a/training/conf/datamodule/transform/paragraphs.yaml
+++ /dev/null
@@ -1,30 +0,0 @@
-# random_crop:
-# _target_: torchvision.transforms.RandomCrop
-# size: [576, 640]
-# padding: null
-# pad_if_needed: true
-# fill: 0
-# padding_mode: constant
-
-center_crop:
- _target_: torchvision.transforms.CenterCrop
- size: [576, 640]
-
-color_jitter:
- _target_: torchvision.transforms.ColorJitter
- brightness: [0.8, 1.6]
-
-random_affine:
- _target_: torchvision.transforms.RandomAffine
- degrees: 1
- shear: [-5, 5]
- interpolation: BILINEAR
-
-random_perspective:
- _target_: torchvision.transforms.RandomPerspective
- distortion_scale: 0.05
- p: 0.25
- fill: 0
-
-to_tensor:
- _target_: torchvision.transforms.ToTensor