颜色分类leetcode StarGAN Tensorflow StarGAN在Tensorflow中的实现
颜色分类leetcode StarGAN-Tensorflow StarGAN在Tensorflow中的实现这段代码主要是从goldkim92的官方pytorch代码的基础上修改而来的。修改代码,使其与官方实现更加一致。修复丢失计算中的错误。增加了更多的测试功能。添加残差块(基于)先决条件Python 3.5张量流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。您也可以尝试不同的方法,例如
用户评论