【文件属性】:
文件名称:颜色分类leetcode-StarGAN-Tensorflow:StarGAN在Tensorflow中的实现
文件大小:2.32MB
文件格式:ZIP
更新时间:2021-07-06 23:07:20
系统开源
颜色分类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)