summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/transform
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-05 19:27:34 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-05 19:27:34 +0100
commit194be7aa14744c2b5a9ef4e31ca19fb70ddfd775 (patch)
treedd329cb14c6beae6dc8dedf409b415c3e8e5edee /training/conf/datamodule/transform
parentbc5a3f16f61fc3d53c87f5aebbe4bafdcc634119 (diff)
Add warp transform to paragraphs
Diffstat (limited to 'training/conf/datamodule/transform')
-rw-r--r--training/conf/datamodule/transform/paragraphs.yaml6
1 files changed, 6 insertions, 0 deletions
diff --git a/training/conf/datamodule/transform/paragraphs.yaml b/training/conf/datamodule/transform/paragraphs.yaml
index d33a4c3..af06c99 100644
--- a/training/conf/datamodule/transform/paragraphs.yaml
+++ b/training/conf/datamodule/transform/paragraphs.yaml
@@ -16,5 +16,11 @@ random_affine:
shear: [-10, 10]
interpolation: BILINEAR
+random_perspective:
+ _target_: torchvision.transforms.RandomPerspective
+ distortion_scale: 0.15
+ p: 0.5
+ fill: 0
+
to_tensor:
_target_: torchvision.transforms.ToTensor