文件名称:颜色分类leetcode-StarGAN-Tensorflow:StarGAN在Tensorflow中的实现
文件大小:2.32MB
文件格式:ZIP
更新时间:2024-07-26 16:54:00
系统开源
颜色分类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。 您也可以尝试不同的方法,例如
【文件预览】:
StarGAN-Tensorflow-master
----util.py(4KB)
----download.py(4KB)
----ops.py(2KB)
----img()
--------example1.png(2.31MB)
----main.py(4KB)
----model.py(18KB)
----README.md(3KB)
----module.py(5KB)
----.gitignore(1KB)