Skip to content

Commit 0d1ef2d

Browse files
committed
set the input size of segformer_b0 as 1024*1024
1 parent 842c20c commit 0d1ef2d

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

Diff for: benchmark/configs/segformer_b0.yml

+15
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,18 @@ loss:
2424
types:
2525
- type: CrossEntropyLoss
2626
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

Comments
 (0)