|
1 | 1 | # 准备 Composition-1k 数据集
|
2 | 2 |
|
| 3 | +## 介绍 |
| 4 | + |
3 | 5 | <!-- [DATASET] -->
|
4 | 6 |
|
5 | 7 | ```bibtex
|
|
11 | 13 | year={2017}
|
12 | 14 | }
|
13 | 15 | ```
|
| 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