diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-05 19:27:34 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-05 19:27:34 +0100 |
| commit | 194be7aa14744c2b5a9ef4e31ca19fb70ddfd775 (patch) | |
| tree | dd329cb14c6beae6dc8dedf409b415c3e8e5edee /training | |
| parent | bc5a3f16f61fc3d53c87f5aebbe4bafdcc634119 (diff) | |
Add warp transform to paragraphs
Diffstat (limited to 'training')
| -rw-r--r-- | training/conf/datamodule/transform/paragraphs.yaml | 6 |
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 |