【文件属性】:
文件名称:cyclegan
文件大小:44KB
文件格式:ZIP
更新时间:2021-03-17 21:00:07
Python
循环GAN-TensorFlow
使用TensorFlow进行CycleGan的实现(正在进行中)。
原始论文: :
测试数据结果
苹果->橙色
输入
输出
输入
输出
输入
输出
橙色->苹果
输入
输出
输入
输出
输入
输出
环境
TensorFlow 1.0.0
的Python 3.6.0
数据准备
首先,下载一个数据集,例如apple2orange
$ bash download_dataset.sh apple2orange
将数据集写入tfrecords
$ python3 build_data.py
检查$ python3 build_data.py --help以获得更多详细信息。
训练
$ python3 train.py
如果要更改某些默认设置,可以将其传递给命令行,例如:
$ python3 train.py \
--X=data/t
【文件预览】:
cyclegan-main
----.gitignore(1KB)
----README.md(5KB)
----discriminator.py(2KB)
----export_graph.py(2KB)
----ops.py(7KB)
----train.py(5KB)
----LICENSE(1KB)
----build_data.py(3KB)
----inference.py(2KB)
----utils.py(1KB)
----model.py(7KB)
----checkpoints()
--------temp(5B)
----__pycache__()
--------ops.cpython-36.pyc(7KB)
--------reader.cpython-37.pyc(3KB)
--------generator.cpython-36.pyc(2KB)
--------generator.cpython-37.pyc(2KB)
--------ops.cpython-37.pyc(7KB)
--------utils.cpython-37.pyc(2KB)
--------model.cpython-37.pyc(5KB)
--------discriminator.cpython-37.pyc(1KB)
--------reader.cpython-36.pyc(3KB)
--------utils.cpython-36.pyc(2KB)
--------model.cpython-36.pyc(5KB)
--------discriminator.cpython-36.pyc(1KB)
----.idea()
--------.gitignore(50B)
--------workspace.xml(6KB)
--------misc.xml(203B)
--------modules.xml(311B)
--------CycleGAN-TensorFlow-master.iml(587B)
--------inspectionProfiles()
----reader.py(3KB)
----download_dataset.sh(891B)
----generator.py(2KB)
----data()
--------tfrecords()
----Makefile(1KB)