We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 842c20c commit 0d1ef2dCopy full SHA for 0d1ef2d
benchmark/configs/segformer_b0.yml
@@ -24,3 +24,18 @@ loss:
24
types:
25
- type: CrossEntropyLoss
26
coef: [1]
27
+
28
+train_dataset:
29
+ transforms:
30
+ - type: ResizeStepScaling
31
+ min_scale_factor: 0.5
32
+ max_scale_factor: 2.0
33
+ scale_step_size: 0.25
34
+ - type: RandomPaddingCrop
35
+ crop_size: [1024, 1024]
36
+ - type: RandomHorizontalFlip
37
+ - type: RandomDistort
38
+ brightness_range: 0.4
39
+ contrast_range: 0.4
40
+ saturation_range: 0.4
41
+ - type: Normalize
0 commit comments