注意
您正在阅读 MMEditing 0.x。 MMEditing 0.x 会在 2022 年末开始逐步停止维护,建议您及时升级到 MMEditing 1.0 版本,享受由 OpenMMLab 2.0 带来的更多新特性和更佳的性能表现。阅读 MMEditing 1.0 的发版日志、 代码 和 文档 以了解更多。
图像生成数据集¶
建议将数据集软链接到 $MMEDITING/data
。如果您的文件夹结构不同,您可能需要更改配置文件中的相应路径。
MMEditing 支持的生成数据集:
为 Pix2pix 准备配对数据集¶
@inproceedings{isola2017image,
title={Image-to-image translation with conditional adversarial networks},
author={Isola, Phillip and Zhu, Jun-Yan and Zhou, Tinghui and Efros, Alexei A},
booktitle={Proceedings of the IEEE conference on computer vision and pattern recognition},
pages={1125--1134},
year={2017}
}
您可以从此处下载配对数据集。然后,您需要解压缩并移动相应的数据集以遵循如下所示的文件夹结构。数据集已经由原作者准备好了。
mmediting
├── mmedit
├── tools
├── configs
├── data
│ ├── paired
│ │ ├── facades
│ │ ├── maps
| | ├── edges2shoes
| | | ├── train
| | | ├── test
为 CycleGAN 准备未配对数据集¶
@inproceedings{zhu2017unpaired,
title={Unpaired image-to-image translation using cycle-consistent adversarial networks},
author={Zhu, Jun-Yan and Park, Taesung and Isola, Phillip and Efros, Alexei A},
booktitle={Proceedings of the IEEE international conference on computer vision},
pages={2223--2232},
year={2017}
}
您可以从此处下载未配对的数据集。然后,您需要解压缩并移动相应的数据集以遵循如上所示的文件夹结构。数据集已经由原作者准备好了。
mmediting
├── mmedit
├── tools
├── configs
├── data
│ ├── unpaired
│ │ ├── facades
| | ├── horse2zebra
| | ├── summer2winter_yosemite
| | | ├── trainA
| | | ├── trainB
| | | ├── testA
| | | ├── testB