pix2pix是继DCGAN之后生成对抗网络又一重要研究进展

实验参考代码https://github.com/Eyyub/tensorflow-pix2pix

首先需要下载 CMP Facade 数据集.

将数据集置于项目根目录

运行代码 python build_dataset.py 'CMP_facade_DB_base/base'

运行python example.py

实验环境

GTX960M--2G显存

以下为迭代0次到3000次的效果图。

但是发现D和G收敛的非常慢,原文使用NVIDIA K80,需要1-8个小时才能跑完train