颜色分类leetcode StarGAN-Tensorflow StarGAN在Tensorflow中的实现，这段代码主要是从goldkim92的官方pytorch代码的基础上修改而来的。修改代码，使其与官方实现更加一致。修复丢失计算中的错误，增加了更多的测试功能，添加了残差块。

先决条件

  • Python 3.5

  • TensorFlow 1.3.0

  • tqdm

用法

只实现了CelebA部分。

首先，下载数据集：


$ python download.py

训练模型：


$ python main.py --phase=train --image_size=64 --batch_size=16

默认的分类方法是使用sigmoid。如果您选择的属性满足单一属性分类（例如：仅头发颜色），或者如果您可以访问RAFD，您也可以尝试使用softmax。


$ python main.py --phase=train --image_size=64 --batch_size=16 --c_method=Softmax

默认的对抗训练方法是改进的WGAN。