Skip to content

Commit 602a040

Browse files
authored
[Doc] Chinese translation of Comp1k README (open-mmlab#583)
1 parent f70bf3b commit 602a040

File tree

1 file changed

+119
-0
lines changed

1 file changed

+119
-0
lines changed

tools/data/matting/comp1k/README_zh-CN.md

+119
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
# 准备 Composition-1k 数据集
22

3+
## 介绍
4+
35
<!-- [DATASET] -->
46

57
```bibtex
@@ -11,3 +13,120 @@
1113
year={2017}
1214
}
1315
```
16+
17+
Adobe Composition-1k 数据集由前景图像及其相应的 alpha 图像组成。要获得完整的数据集,需要将前景与来自 COCO 数据集和 Pascal VOC 数据集的选定背景进行合成。
18+
19+
## 获取和提取
20+
21+
请按照 [论文作者](https://sites.google.com/view/deepimagematting) 的说明获取 Composition-1k (comp1k) 数据集。
22+
23+
## 合成完整数据集
24+
25+
Adobe composition-1k 数据集仅包含 `alpha``fg`(以及测试集中的 `trimap`)。在训练或评估之前,需要将 `fg` 与 COCO 数据(训练)或 VOC 数据(测试)合并。使用以下脚本执行图像合成并生成用于训练或测试的注释文件:
26+
27+
```shell
28+
# 在 MMEditing 的根文件夹下运行脚本
29+
python tools/data/matting/comp1k/preprocess_comp1k_dataset.py data/adobe_composition-1k data/coco data/VOCdevkit --composite
30+
```
31+
32+
生成的数据分别存储在 “adobe_composition-1k/Training_set” 和 “adobe_composition-1k/Test_set” 下。如果你只想合成测试数据(因为合成训练数据很耗时),你可以通过删除 `--composite` 选项来跳过合成训练集:
33+
34+
```shell
35+
# 跳过合成训练集
36+
python tools/data/matting/comp1k/preprocess_comp1k_dataset.py data/adobe_composition-1k data/coco data/VOCdevkit
37+
```
38+
39+
如果您只想预处理测试数据,即对于 FBA,您可以通过添加 `--skip_train` 选项来跳过训练集:
40+
41+
```shell
42+
# 跳过预处理训练集
43+
python tools/data/matting/comp1k/preprocess_comp1k_dataset.py data/adobe_composition-1k data/coco data/VOCdevkit --skip_train
44+
```
45+
46+
> 目前,`GCA``FBA` 支持在线合成训练数据。但是你可以修改其他模型的数据管道来执行在线合成,而不是加载合成图像(我们在数据管道中称之为“合并”)。
47+
48+
## 检查 DIM 的目录结构
49+
50+
最终的文件夹结构应如下所示:
51+
52+
```text
53+
mmediting
54+
├── mmedit
55+
├── tools
56+
├── configs
57+
├── data
58+
│ ├── adobe_composition-1k
59+
│ │ ├── Test_set
60+
│ │ │ ├── Adobe-licensed images
61+
│ │ │ │ ├── alpha
62+
│ │ │ │ ├── fg
63+
│ │ │ │ ├── trimaps
64+
│ │ │ ├── merged (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
65+
│ │ │ ├── bg (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
66+
│ │ ├── Training_set
67+
│ │ │ ├── Adobe-licensed images
68+
│ │ │ │ ├── alpha
69+
│ │ │ │ ├── fg
70+
│ │ │ ├── Other
71+
│ │ │ │ ├── alpha
72+
│ │ │ │ ├── fg
73+
│ │ │ ├── merged (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
74+
│ │ │ ├── bg (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
75+
│ │ ├── test_list.json (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
76+
│ │ ├── training_list.json (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
77+
│ ├── coco
78+
│ │ ├── train2014 (or train2017)
79+
│ ├── VOCdevkit
80+
│ │ ├── VOC2012
81+
```
82+
## 为 FBA 准备数据集
83+
84+
FBA采用 [Learning-base Sampling for Natural Image Matting](https://openaccess.thecvf.com/content_CVPR_2019/papers/Tang_Learning-Based_Sampling_for_Natural_Image_Matting_CVPR_2019_paper.pdf) 中提出的动态数据集增强。此外,为了减少增强过程中的伪影,它使用前景的扩展版本作为前景。我们提供脚本来估计前景。
85+
86+
准备测试集:
87+
88+
```shell
89+
# 跳过预处理训练集,因为它在训练期间在线合成
90+
python tools/data/matting/comp1k/preprocess_comp1k_dataset.py data/adobe_composition-1k data/coco data/VOCdevkit --skip_train
91+
```
92+
93+
扩展训练集的前景:
94+
95+
```shell
96+
python tools/data/matting/comp1k/extend_fg.py data/adobe_composition-1k
97+
```
98+
99+
## 检查 DIM 的目录结构
100+
101+
最终的文件夹结构应如下所示:
102+
103+
```text
104+
mmediting
105+
├── mmedit
106+
├── tools
107+
├── configs
108+
├── data
109+
│ ├── adobe_composition-1k
110+
│ │ ├── Test_set
111+
│ │ │ ├── Adobe-licensed images
112+
│ │ │ │ ├── alpha
113+
│ │ │ │ ├── fg
114+
│ │ │ │ ├── trimaps
115+
│ │ │ ├── merged (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
116+
│ │ │ ├── bg (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
117+
│ │ ├── Training_set
118+
│ │ │ ├── Adobe-licensed images
119+
│ │ │ │ ├── alpha
120+
│ │ │ │ ├── fg
121+
│ │ │ │ ├── fg_extended (generated by tools/data/matting/comp1k/extend_fg.py)
122+
│ │ │ ├── Other
123+
│ │ │ │ ├── alpha
124+
│ │ │ │ ├── fg
125+
│ │ │ │ ├── fg_extended (generated by tools/data/matting/comp1k/extend_fg.py)
126+
│ │ ├── test_list.json (generated by tools/data/matting/comp1k/preprocess_comp1k_dataset.py)
127+
│ │ ├── training_list_fba.json (generated by tools/data/matting/comp1k/extend_fg.py)
128+
│ ├── coco
129+
│ │ ├── train2014 (or train2017)
130+
│ ├── VOCdevkit
131+
│ │ ├── VOC2012
132+
```

0 commit comments

Comments
 (0)