diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-11 22:51:31 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-11 22:51:31 +0100 |
| commit | 94ec29dc00b8e19dd7fcdaac0856b6cf00abd469 (patch) | |
| tree | dc7d08f997bbde1d5d3f58051d7b6bce006738e8 /training/conf/datamodule | |
| parent | adc36f7892ffedeafed7579282d62fbeecd496e9 (diff) | |
fix(paragraphs-conf): replace random crop with center crop
Diffstat (limited to 'training/conf/datamodule')
| -rw-r--r-- | training/conf/datamodule/transform/paragraphs.yaml | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/training/conf/datamodule/transform/paragraphs.yaml b/training/conf/datamodule/transform/paragraphs.yaml index 1b0ab04..5e5314c 100644 --- a/training/conf/datamodule/transform/paragraphs.yaml +++ b/training/conf/datamodule/transform/paragraphs.yaml @@ -1,10 +1,14 @@ -random_crop: - _target_: torchvision.transforms.RandomCrop +# random_crop: +# _target_: torchvision.transforms.RandomCrop +# size: [576, 640] +# padding: null +# pad_if_needed: true +# fill: 0 +# padding_mode: constant + +center_crop: + _target_: torchvision.transforms.CenterCrop size: [576, 640] - padding: null - pad_if_needed: true - fill: 0 - padding_mode: constant color_jitter: _target_: torchvision.transforms.ColorJitter |