| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- batch_size: 4
- iters: 300000
- train_dataset:
- type: MattingDataset
- dataset_root: data/matting/Distinctions-646
- train_file: train.txt
- transforms:
- - type: LoadImages
- - type: Padding
- target_size: [512, 512]
- - type: RandomCrop
- crop_size: [[512, 512],[640, 640], [800, 800]]
- - type: Resize
- target_size: [512, 512]
- - type: RandomDistort
- - type: RandomBlur
- prob: 0.1
- - type: RandomHorizontalFlip
- - type: Normalize
- mode: train
- separator: '|'
- val_dataset:
- type: MattingDataset
- dataset_root: data/matting/Distinctions-646
- val_file: val.txt
- transforms:
- - type: LoadImages
- - type: LimitShort
- max_short: 1536
- - type: ResizeToIntMult
- mult_int: 32
- - type: Normalize
- mode: val
- get_trimap: False
- separator: '|'
- model:
- type: PPMatting
- backbone:
- type: HRNet_W48
- pretrained: https://bj.bcebos.com/paddleseg/dygraph/hrnet_w48_ssld.tar.gz
- pretrained: Null
- optimizer:
- type: sgd
- momentum: 0.9
- weight_decay: 4.0e-5
- lr_scheduler:
- type: PolynomialDecay
- learning_rate: 0.01
- end_lr: 0
- power: 0.9
|
