From 194be7aa14744c2b5a9ef4e31ca19fb70ddfd775 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm 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(+) 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