| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647 |
- batch_size: 1
- iters: 1000
- train_dataset:
- type: MattingDataset
- dataset_root: data/PPM-100
- train_file: train.txt
- transforms:
- - type: LoadImages
- - type: RandomCrop
- crop_size: [512, 512]
- - type: RandomDistort
- - type: RandomBlur
- - type: RandomHorizontalFlip
- - type: Normalize
- mode: train
- val_dataset:
- type: MattingDataset
- dataset_root: data/PPM-100
- val_file: val.txt
- transforms:
- - type: LoadImages
- - type: ResizeByShort
- short_size: 512
- - type: ResizeToIntMult
- mult_int: 32
- - type: Normalize
- mode: val
- get_trimap: False
- model:
- type: MODNet
- backbone:
- type: MobileNetV2
- pretrained: https://paddleseg.bj.bcebos.com/matting/models/MobileNetV2_pretrained/model.pdparams
- pretrained: Null
- optimizer:
- type: sgd
- momentum: 0.9
- weight_decay: 4.0e-5
- lr_scheduler:
- type: PiecewiseDecay
- boundaries: [40000, 80000]
- values: [0.02, 0.002, 0.0002]
|
