summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/transform/lines.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-05 19:23:48 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-05 19:24:39 +0100
commitf53a7eb596ecfce0d36b5f5cb455dad6aff0360b (patch)
tree86801deea7086191988dedf25d3d573ccaf7ca80 /training/conf/datamodule/transform/lines.yaml
parent40dd50690937cb3f8adb644363b4830ecd7fcfcc (diff)
Rename iam lines transforms
Rename iam lines transform
Diffstat (limited to 'training/conf/datamodule/transform/lines.yaml')
-rw-r--r--training/conf/datamodule/transform/lines.yaml23
1 files changed, 23 insertions, 0 deletions
diff --git a/training/conf/datamodule/transform/lines.yaml b/training/conf/datamodule/transform/lines.yaml
new file mode 100644
index 0000000..6949a15
--- /dev/null
+++ b/training/conf/datamodule/transform/lines.yaml
@@ -0,0 +1,23 @@
+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: [-30, 20]
+ interpolation: BILINEAR
+ fill: 0
+
+random_perspective:
+ _target_: torchvision.transforms.RandomPerspective
+ distortion_scale: 0.2
+ p: 0.5
+ fill: 0
+
+to_tensor:
+ _target_: torchvision.transforms.ToTensor