summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/transform/iam_lines_barlow.yaml
blob: b44ca4e5315f37bb30b0a05160a150235edab0d9 (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
barlow:
  _target_: text_recognizer.data.transforms.barlow.BarlowTransform
  prim:
    embed_crop:
      _target_: text_recognizer.data.transforms.embed_crop.EmbedCrop
      augment: true

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

    random_affine:
      _target_: torchvision.transforms.RandomAffine
      degrees: 1
      shear: [-30, 20]
      interpolation: BILINEAR
      fill: 0

    to_tensor:
      _target_: torchvision.transforms.ToTensor

  bis:
    embed_crop:
      _target_: text_recognizer.data.transforms.embed_crop.EmbedCrop
      augment: true

    color_jitter:
      _target_: torchvision.transforms.ColorJitter
      brightness: [0.5, 1.4]

    random_affine:
      _target_: torchvision.transforms.RandomAffine
      degrees: 1.5
      shear: [-30, 20]
      interpolation: BILINEAR
      fill: 0

    to_tensor:
      _target_: torchvision.transforms.ToTensor