|
batch_size: 16 |
|
iters: 100000 |
|
|
|
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: 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] |
|
|