# 模型列表

## 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
