benchmark_cn.md 1.4 KB
Newer Older
W
wuyefeilin 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
# 模型列表

## 1. 训练Benchmark

### 1.1 软硬件环境

* PP-Matting训练过程中使用单卡GPU,batch size为4。

### 1.2 数据集

* 通用目标抠图数据集为Compositon-1k或Distinctions-646(使用该两者数据集需向作者进行申请),使用COCO2017和Pascal VOC 2012作为背景数据集。
* 人像抠图使用内部数据。

### 1.3 指标
|模型名称 | 模型简介 | 输入尺寸 |
|---|---|---|
|ppmatting_hrnet_w48 | 通用目标抠图 | 512 |
|ppmatting_hrnet_w18 | 人像抠图 | 512 |

## 2. 推理Benchmark

### 2.1 软硬件环境

* 模型推理速度测试采用单卡V100,batch size=1进行测试,使用CUDA 10.2, CUDNN 7.6.5, PaddlePaddle-gpu 2.3.2。

### 2.2 数据集
* 通用目标抠图:Compositon-1k或Distinctions-646中的测试集部分。
* 人像抠图:PPM-100和AIM-500中的人像部分,共195张, 记为PPM-AIM-195。

### 2.3 指标
| 模型 | 数据集 | SAD | MSE | Grad | Conn |Params(M) | FLOPs(G) | FPS |
| - | - | -| - | - | - | - | -| - | - |
| ppmatting_hrnet_w48 | Composition-1k | 46.22 | 0.005 | 22.69 | 45.40 | 86.3 | 165.4 | 24.4 |
| ppmatting_hrnet_w48 | Distinctions-646 | 40.69 | 0.009 | 43.91 |40.56 | 86.3 | 165.4 | 24.4 |
| ppmatting_hrnet_w18 | PPM-AIM-195 | 31.56|0.0022|31.80|30.13| 24.5 | 91.28 | 28.9 |

## 3. 相关使用说明
1. https://github.com/PaddlePaddle/PaddleSeg/tree/develop/Matting