本库用 TensorFlow 简单的实现了 StarGAN。
StarGAN 是一种新颖且可扩展的方法,可以仅使用一个模型来执行多个域的图像到图像的转换。StarGAN 这样一个统一的模型体系架构让开发者可以同时训练单个网络中具有不同域的多个数据集,这导致StarGAN的图像转化结果比现有模型质量更高,并具有将输入图像灵活转化成任何期望目标域的新颖能力。
Github:
https://github.com/taki0112/StarGAN-Tensorflow
下载数据集:
> python download.py celebA
├── dataset └── celebA ├── train ├── 000001.jpg ├── 000002.jpg └── ... ├── test (It is not celebA) ├── a.jpg (The test image that you wanted) ├── b.png └── ... ├── list_attr_celeba.txt (For attribute information)
训练:
测试:
预训练模型:
女人
男人