summaryrefslogtreecommitdiff
path: root/training/conf
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf')
-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