Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
UNet_paddle
Commits
0d97cc8c
Commit
0d97cc8c
authored
Jun 07, 2023
by
Sugon_ldc
Browse files
add new model
parents
Pipeline
#316
failed with stages
in 0 seconds
Changes
613
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
544 additions
and
0 deletions
+544
-0
Matting/configs/benchmarks/PPM/README.md
Matting/configs/benchmarks/PPM/README.md
+9
-0
Matting/configs/benchmarks/PPM/closeform.yml
Matting/configs/benchmarks/PPM/closeform.yml
+19
-0
Matting/configs/benchmarks/PPM/fast.yml
Matting/configs/benchmarks/PPM/fast.yml
+6
-0
Matting/configs/benchmarks/PPM/knn.yml
Matting/configs/benchmarks/PPM/knn.yml
+6
-0
Matting/configs/benchmarks/PPM/learningbased.yml
Matting/configs/benchmarks/PPM/learningbased.yml
+6
-0
Matting/configs/benchmarks/PPM/randomwalks.yml
Matting/configs/benchmarks/PPM/randomwalks.yml
+6
-0
Matting/configs/dim/dim-vgg16.yml
Matting/configs/dim/dim-vgg16.yml
+45
-0
Matting/configs/human_matting/human_matting-resnet34_vd.yml
Matting/configs/human_matting/human_matting-resnet34_vd.yml
+54
-0
Matting/configs/modnet/modnet-hrnet_w18.yml
Matting/configs/modnet/modnet-hrnet_w18.yml
+5
-0
Matting/configs/modnet/modnet-mobilenetv2.yml
Matting/configs/modnet/modnet-mobilenetv2.yml
+47
-0
Matting/configs/modnet/modnet-resnet50_vd.yml
Matting/configs/modnet/modnet-resnet50_vd.yml
+5
-0
Matting/configs/ppmatting/README.md
Matting/configs/ppmatting/README.md
+20
-0
Matting/configs/ppmatting/ppmatting-hrnet_w18-human_1024.yml
Matting/configs/ppmatting/ppmatting-hrnet_w18-human_1024.yml
+29
-0
Matting/configs/ppmatting/ppmatting-hrnet_w18-human_512.yml
Matting/configs/ppmatting/ppmatting-hrnet_w18-human_512.yml
+44
-0
Matting/configs/ppmatting/ppmatting-hrnet_w48-composition.yml
...ing/configs/ppmatting/ppmatting-hrnet_w48-composition.yml
+8
-0
Matting/configs/ppmatting/ppmatting-hrnet_w48-distinctions.yml
...ng/configs/ppmatting/ppmatting-hrnet_w48-distinctions.yml
+56
-0
Matting/configs/ppmattingv2/ppmattingv2-stdc1-human_512.yml
Matting/configs/ppmattingv2/ppmattingv2-stdc1-human_512.yml
+66
-0
Matting/configs/quick_start/modnet-mobilenetv2.yml
Matting/configs/quick_start/modnet-mobilenetv2.yml
+47
-0
Matting/configs/quick_start/ppmattingv2-stdc1-human_512.yml
Matting/configs/quick_start/ppmattingv2-stdc1-human_512.yml
+66
-0
Matting/demo/human.jpg
Matting/demo/human.jpg
+0
-0
No files found.
Too many changes to show.
To preserve performance only
613 of 613+
files are displayed.
Plain diff
Email patch
Matting/configs/benchmarks/PPM/README.md
0 → 100644
View file @
0d97cc8c
### PPM
| Method | SAD | MSE | Grad | Conn |
|-|-|-|-|-|
|ClosedFormMatting|40.6251|0.0782|55.5716|40.6646|
|KNNMatting|41.5604|0.0681|52.5200|42.1784|
|FastMatting|35.8735|0.0492|48.9267|35.6183|
|LearningBasedMatting|40.5506|0.0776|55.3923|40.5690|
|RandomWalksMatting|54.6315|0.0962|69.8779|54.0870|
Matting/configs/benchmarks/PPM/closeform.yml
0 → 100644
View file @
0d97cc8c
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
:
True
model
:
type
:
CloseFormMatting
Matting/configs/benchmarks/PPM/fast.yml
0 → 100644
View file @
0d97cc8c
_base_
:
closeform.yml
model
:
type
:
FastMatting
Matting/configs/benchmarks/PPM/knn.yml
0 → 100644
View file @
0d97cc8c
_base_
:
closeform.yml
model
:
type
:
KNNMatting
Matting/configs/benchmarks/PPM/learningbased.yml
0 → 100644
View file @
0d97cc8c
_base_
:
closeform.yml
model
:
type
:
LearningBasedMatting
Matting/configs/benchmarks/PPM/randomwalks.yml
0 → 100644
View file @
0d97cc8c
_base_
:
closeform.yml
model
:
type
:
RandomWalksMatting
Matting/configs/dim/dim-vgg16.yml
0 → 100644
View file @
0d97cc8c
batch_size
:
16
iters
:
100000
train_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
train_file
:
train.txt
transforms
:
-
type
:
LoadImages
-
type
:
RandomCropByAlpha
crop_size
:
[[
320
,
320
],
[
480
,
480
],
[
640
,
640
]]
-
type
:
Resize
target_size
:
[
320
,
320
]
-
type
:
RandomDistort
-
type
:
RandomBlur
-
type
:
RandomHorizontalFlip
-
type
:
Normalize
mode
:
train
get_trimap
:
True
val_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
val_file
:
val.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitLong
max_long
:
3840
-
type
:
Normalize
mode
:
val
get_trimap
:
True
model
:
type
:
DIM
backbone
:
type
:
VGG16
input_channels
:
4
pretrained
:
https://paddleseg.bj.bcebos.com/matting/models/DIM_VGG16_pretrained/model.pdparams
pretrained
:
Null
optimizer
:
type
:
adam
learning_rate
:
value
:
0.001
Matting/configs/human_matting/human_matting-resnet34_vd.yml
0 → 100644
View file @
0d97cc8c
batch_size
:
4
iters
:
50000
train_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
train_file
:
train.txt
transforms
:
-
type
:
LoadImages
-
type
:
RandomResize
size
:
[
2048
,
2048
]
scale
:
[
0.3
,
1.5
]
-
type
:
RandomCrop
crop_size
:
[
2048
,
2048
]
-
type
:
RandomDistort
-
type
:
RandomBlur
prob
:
0.1
-
type
:
RandomHorizontalFlip
-
type
:
Padding
target_size
:
[
2048
,
2048
]
-
type
:
Normalize
mode
:
train
val_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
val_file
:
val.txt
transforms
:
-
type
:
LoadImages
-
type
:
ResizeByShort
short_size
:
2048
-
type
:
ResizeToIntMult
mult_int
:
128
-
type
:
Normalize
mode
:
val
get_trimap
:
False
model
:
type
:
HumanMatting
backbone
:
type
:
ResNet34_vd
pretrained
:
https://paddleseg.bj.bcebos.com/matting/models/ResNet34_vd_pretrained/model.pdparams
pretrained
:
Null
if_refine
:
True
optimizer
:
type
:
sgd
momentum
:
0.9
weight_decay
:
4.0e-5
lr_scheduler
:
type
:
PiecewiseDecay
boundaries
:
[
30000
,
40000
]
values
:
[
0.001
,
0.0001
,
0.00001
]
Matting/configs/modnet/modnet-hrnet_w18.yml
0 → 100644
View file @
0d97cc8c
_base_
:
modnet-mobilenetv2.yml
model
:
backbone
:
type
:
HRNet_W18
pretrained
:
https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz
Matting/configs/modnet/modnet-mobilenetv2.yml
0 → 100644
View file @
0d97cc8c
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
:
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
]
Matting/configs/modnet/modnet-resnet50_vd.yml
0 → 100644
View file @
0d97cc8c
_base_
:
modnet-mobilenetv2.yml
model
:
backbone
:
type
:
ResNet50_vd
pretrained
:
https://bj.bcebos.com/paddleseg/dygraph/resnet50_vd_ssld_v2.tar.gz
Matting/configs/ppmatting/README.md
0 → 100644
View file @
0d97cc8c
# PP-Matting: High-Accuracy Natural Image Matting
## Reference
> Chen G, Liu Y, Wang J, et al. PP-Matting: High-Accuracy Natural Image Matting[J]. arXiv preprint arXiv:2204.09433, 2022.
## Performance
### Composition-1k
| Model | Backbone | Resolution | Training Iters | SAD $
\d
ownarrow$ | MSE $
\d
ownarrow$ | Grad $
\d
ownarrow$ | Conn $
\d
ownarrow$ | Links |
|:-:|:-:|:-:|:-:|:-:|:-:|:-:|:-:|:-:|
|PP-Matting|HRNet_W48|512x512|300000|46.22|0.005|22.69|45.40|
[
model
](
https://paddleseg.bj.bcebos.com/matting/models/ppmatting-hrnet_w48-composition.pdparams
)
|
### Distinctions-646
| Model | Backbone | Resolution | Training Iters | SAD $
\d
ownarrow$ | MSE $
\d
ownarrow$ | Grad $
\d
ownarrow$ | Conn $
\d
ownarrow$ | Links |
|:-:|:-:|:-:|:-:|:-:|:-:|:-:|:-:|:-:|
|PP-Matting|HRNet_W48|512x512|300000|40.69|0.009|43.91|40.56|
[
model
](
https://paddleseg.bj.bcebos.com/matting/models/ppmatting-hrnet_w48-distinctions.pdparams
)
|
Matting/configs/ppmatting/ppmatting-hrnet_w18-human_1024.yml
0 → 100644
View file @
0d97cc8c
_base_
:
'
ppmatting-hrnet_w18-human_512.yml'
train_dataset
:
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
1024
-
type
:
RandomCrop
crop_size
:
[
1024
,
1024
]
-
type
:
RandomDistort
-
type
:
RandomBlur
prob
:
0.1
-
type
:
RandomNoise
prob
:
0.5
-
type
:
RandomReJpeg
prob
:
0.2
-
type
:
RandomHorizontalFlip
-
type
:
Normalize
val_dataset
:
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
1024
-
type
:
ResizeToIntMult
mult_int
:
32
-
type
:
Normalize
Matting/configs/ppmatting/ppmatting-hrnet_w18-human_512.yml
0 → 100644
View file @
0d97cc8c
_base_
:
'
ppmatting-hrnet_w48-distinctions.yml'
batch_size
:
4
iters
:
200000
train_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
train_file
:
train.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
512
-
type
:
RandomCrop
crop_size
:
[
512
,
512
]
-
type
:
RandomDistort
-
type
:
RandomBlur
prob
:
0.1
-
type
:
RandomNoise
prob
:
0.5
-
type
:
RandomReJpeg
prob
:
0.2
-
type
:
RandomHorizontalFlip
-
type
:
Normalize
mode
:
train
val_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
val_file
:
val.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
512
-
type
:
ResizeToIntMult
mult_int
:
32
-
type
:
Normalize
mode
:
val
get_trimap
:
False
model
:
backbone
:
type
:
HRNet_W18
pretrained
:
https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz
Matting/configs/ppmatting/ppmatting-hrnet_w48-composition.yml
0 → 100644
View file @
0d97cc8c
_base_
:
'
ppmatting-hrnet_w48-distinctions.yml'
train_dataset
:
dataset_root
:
data/matting/Composition-1k
val_dataset
:
dataset_root
:
data/matting/Composition-1k
\ No newline at end of file
Matting/configs/ppmatting/ppmatting-hrnet_w48-distinctions.yml
0 → 100644
View file @
0d97cc8c
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
\ No newline at end of file
Matting/configs/ppmattingv2/ppmattingv2-stdc1-human_512.yml
0 → 100644
View file @
0d97cc8c
batch_size
:
16
# total batch size: 16
iters
:
100000
train_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
train_file
:
train.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
512
-
type
:
RandomCrop
crop_size
:
[
512
,
512
]
-
type
:
Padding
target_size
:
[
512
,
512
]
-
type
:
RandomDistort
-
type
:
RandomBlur
prob
:
0.1
-
type
:
RandomSharpen
prob
:
0.2
-
type
:
RandomNoise
prob
:
0.5
-
type
:
RandomReJpeg
prob
:
0.2
-
type
:
RandomHorizontalFlip
-
type
:
Normalize
mode
:
train
val_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
val_file
:
val.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
512
-
type
:
ResizeToIntMult
mult_int
:
32
-
type
:
Normalize
mode
:
val
get_trimap
:
False
model
:
type
:
PPMattingV2
backbone
:
type
:
STDC1
pretrained
:
https://bj.bcebos.com/paddleseg/dygraph/PP_STDCNet1.tar.gz
decoder_channels
:
[
128
,
96
,
64
,
32
,
16
]
head_channel
:
8
dpp_output_channel
:
256
dpp_merge_type
:
add
optimizer
:
type
:
sgd
momentum
:
0.9
weight_decay
:
5.0e-4
lr_scheduler
:
type
:
PolynomialDecay
learning_rate
:
0.01
end_lr
:
0
power
:
0.9
warmup_iters
:
1000
warmup_start_lr
:
1.0e-5
Matting/configs/quick_start/modnet-mobilenetv2.yml
0 → 100644
View file @
0d97cc8c
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
]
Matting/configs/quick_start/ppmattingv2-stdc1-human_512.yml
0 → 100644
View file @
0d97cc8c
batch_size
:
1
iters
:
1000
train_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
train_file
:
train.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
512
-
type
:
RandomCrop
crop_size
:
[
512
,
512
]
-
type
:
Padding
target_size
:
[
512
,
512
]
-
type
:
RandomDistort
-
type
:
RandomBlur
prob
:
0.1
-
type
:
RandomSharpen
prob
:
0.2
-
type
:
RandomNoise
prob
:
0.5
-
type
:
RandomReJpeg
prob
:
0.2
-
type
:
RandomHorizontalFlip
-
type
:
Normalize
mode
:
train
val_dataset
:
type
:
MattingDataset
dataset_root
:
data/PPM-100
val_file
:
val.txt
transforms
:
-
type
:
LoadImages
-
type
:
LimitShort
max_short
:
512
-
type
:
ResizeToIntMult
mult_int
:
32
-
type
:
Normalize
mode
:
val
get_trimap
:
False
model
:
type
:
PPMattingV2
backbone
:
type
:
STDC1
pretrained
:
https://bj.bcebos.com/paddleseg/dygraph/PP_STDCNet1.tar.gz
decoder_channels
:
[
128
,
96
,
64
,
32
,
16
]
head_channel
:
8
dpp_output_channel
:
256
dpp_merge_type
:
add
optimizer
:
type
:
sgd
momentum
:
0.9
weight_decay
:
5.0e-4
lr_scheduler
:
type
:
PolynomialDecay
learning_rate
:
0.01
end_lr
:
0
power
:
0.9
warmup_iters
:
100
warmup_start_lr
:
1.0e-5
Matting/demo/human.jpg
0 → 100644
View file @
0d97cc8c
138 KB
Prev
1
…
7
8
9
10
11
12
13
14
15
…
31
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment