图像生成数据集¶

建议将数据集软链接到 $MMEDITING/data 。如果您的文件夹结构不同，您可能需要更改配置文件中的相应路径。

MMEditing 支持的生成数据集：

Pix2Pix 的配对数据集 [ 主页 ]
CycleGAN 的未配对数据集 [ 主页 ]

为 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