summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-13 19:09:49 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-13 19:09:49 +0200
commit657e9cd4265b5a3111d46911ee5acd27a78a1bf4 (patch)
tree083b5aadf937a26ae692444dadb372eaba1cd013
parent2862fdf77a19c4afa5e00c900af4877df31a3ea6 (diff)
Update transform configs
-rw-r--r--training/conf/datamodule/transform/lines.yaml6
-rw-r--r--training/conf/datamodule/transform/paragraphs.yaml6
2 files changed, 6 insertions, 6 deletions
diff --git a/training/conf/datamodule/transform/lines.yaml b/training/conf/datamodule/transform/lines.yaml
index d7e51e1..6bd35ef 100644
--- a/training/conf/datamodule/transform/lines.yaml
+++ b/training/conf/datamodule/transform/lines.yaml
@@ -9,14 +9,14 @@ color_jitter:
random_affine:
_target_: torchvision.transforms.RandomAffine
degrees: 1
- shear: [-30, 20]
+ shear: [-10, 10]
interpolation: BILINEAR
fill: 0
random_perspective:
_target_: torchvision.transforms.RandomPerspective
- distortion_scale: 0.07
- p: 0.5
+ distortion_scale: 0.05
+ p: 0.25
fill: 0
to_tensor:
diff --git a/training/conf/datamodule/transform/paragraphs.yaml b/training/conf/datamodule/transform/paragraphs.yaml
index 5e5314c..4f260d0 100644
--- a/training/conf/datamodule/transform/paragraphs.yaml
+++ b/training/conf/datamodule/transform/paragraphs.yaml
@@ -17,13 +17,13 @@ color_jitter:
random_affine:
_target_: torchvision.transforms.RandomAffine
degrees: 1
- shear: [-10, 10]
+ shear: [-5, 5]
interpolation: BILINEAR
random_perspective:
_target_: torchvision.transforms.RandomPerspective
- distortion_scale: 0.1
- p: 0.5
+ distortion_scale: 0.05
+ p: 0.25
fill: 0
to_tensor: