PaddleSeg-Matting / configs /modnet /modnet_mobilenetv2.yml
marta-0's picture
add files
6da6215
raw
history blame contribute delete
968 Bytes
batch_size: 16
iters: 100000
train_dataset:
type: MattingDataset
dataset_root: .
train_file: train.txt
transforms:
# - type: LoadImages
- type: ResizeByShort
short_size: 512
- type: RandomCrop
crop_size: [512, 512]
- type: RandomDistort
- type: RandomBlur
- type: RandomHorizontalFlip
- type: Normalize
mode: train
val_dataset:
type: MattingDataset
dataset_root: .
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]