summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/transform/barlow_paragraphs.yaml
blob: 5eefce58e1710435290e9ee2348f19f6ff4bcb69 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

barlow:
  _target_: text_recognizer.data.transforms.barlow.BarlowTransform
  prim:
    random_crop:
      _target_: torchvision.transforms.RandomCrop
      size: [576, 640]
      padding: null
      pad_if_needed: true
      fill: 0
      padding_mode: constant

    color_jitter:
      _target_: torchvision.transforms.ColorJitter
      brightness: [0.8, 1.6]

    random_affine:
      _target_: torchvision.transforms.RandomAffine
      degrees: 1
      shear: [-10, 10]
      interpolation: BILINEAR

    to_tensor:
      _target_: torchvision.transforms.ToTensor

  bis:
    random_crop:
      _target_: torchvision.transforms.RandomCrop
      size: [576, 640]
      padding: null
      pad_if_needed: true
      fill: 0
      padding_mode: constant

    color_jitter:
      _target_: torchvision.transforms.ColorJitter
      brightness: [0.8, 2.0]

    random_affine:
      _target_: torchvision.transforms.RandomAffine
      degrees: 1
      shear: [-5, 5]
      interpolation: BILINEAR

    to_tensor:
      _target_: torchvision.transforms.ToTensor