From 194be7aa14744c2b5a9ef4e31ca19fb70ddfd775 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 5 Nov 2021 19:27:34 +0100
Subject: Add warp transform to paragraphs

---
 training/conf/datamodule/transform/paragraphs.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'training/conf')

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
-- 
cgit v1.2.3-70-g09d2