forked from PaddlePaddle/PaddleSeg
-
Notifications
You must be signed in to change notification settings - Fork 1
/
human_pp_humansegv1_mobile.yml
69 lines (61 loc) · 1.31 KB
/
human_pp_humansegv1_mobile.yml
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
batch_size: 8
iters: 1000
train_dataset:
type: Dataset
dataset_root: data/mini_supervisely
train_path: data/mini_supervisely/train.txt
num_classes: 2
transforms:
- type: Resize
target_size: [192, 192]
- type: ResizeStepScaling
scale_step_size: 0
- type: RandomRotation
- type: RandomPaddingCrop
crop_size: [192, 192]
- type: RandomHorizontalFlip
- type: RandomDistort
- type: RandomBlur
prob: 0.3
- type: Normalize
mode: train
val_dataset:
type: Dataset
dataset_root: data/mini_supervisely
val_path: data/mini_supervisely/val.txt
num_classes: 2
transforms:
- type: Resize
target_size: [192, 192]
- type: Normalize
mode: val
export:
transforms:
- type: Resize
target_size: [192, 192]
- type: Normalize
optimizer:
type: sgd
momentum: 0.9
weight_decay: 0.0005
lr_scheduler:
type: PolynomialDecay
learning_rate: 0.001
end_lr: 0
power: 0.9
loss:
types:
- type: MixedLoss
losses:
- type: CrossEntropyLoss
- type: LovaszSoftmaxLoss
coef: [0.8, 0.2]
coef: [1]
model:
type: FCN
backbone:
type: HRNet_W18_Small_V1
align_corners: False
num_classes: 2
backbone_indices: [-1]
pretrained: pretrained_models/human_pp_humansegv1_mobile_192x192_pretrained/model.pdparams