summaryrefslogtreecommitdiff
path: root/training/conf/datamodule
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
parent40dd50690937cb3f8adb644363b4830ecd7fcfcc (diff)
Rename iam lines transforms
Rename iam lines transform
Diffstat (limited to 'training/conf/datamodule')
-rw-r--r--training/conf/datamodule/iam_lines.yaml4
-rw-r--r--training/conf/datamodule/test_transform/lines.yaml (renamed from training/conf/datamodule/test_transform/iam_lines.yaml)0
-rw-r--r--training/conf/datamodule/transform/lines.yaml (renamed from training/conf/datamodule/transform/iam_lines.yaml)6
3 files changed, 8 insertions, 2 deletions
diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml
index 36e7093..ce4fa91 100644
--- a/training/conf/datamodule/iam_lines.yaml
+++ b/training/conf/datamodule/iam_lines.yaml
@@ -3,5 +3,5 @@ batch_size: 8
num_workers: 12
train_fraction: 0.8
pin_memory: false
-transform: transform/iam_lines.yaml
-test_transform: test_transform/iam_lines.yaml
+transform: transform/lines.yaml
+test_transform: test_transform/lines.yaml
diff --git a/training/conf/datamodule/test_transform/iam_lines.yaml b/training/conf/datamodule/test_transform/lines.yaml
index a14ec82..a14ec82 100644
--- a/training/conf/datamodule/test_transform/iam_lines.yaml
+++ b/training/conf/datamodule/test_transform/lines.yaml
diff --git a/training/conf/datamodule/transform/iam_lines.yaml b/training/conf/datamodule/transform/lines.yaml
index 710f6e3..6949a15 100644
--- a/training/conf/datamodule/transform/iam_lines.yaml
+++ b/training/conf/datamodule/transform/lines.yaml
@@ -13,5 +13,11 @@ random_affine:
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