文件名称:pixel-dt-gan-pytorch:Pytorch像素级域转移的实现
文件大小:60KB
文件格式:ZIP
更新时间:2024-06-03 00:44:59
Python
Piexl-leve域转移的Pytorch实现
参考:
注意(重要):
这段代码不完整,培训还没有结束。
我认为该错误非常琐碎,只要有时间就可以解决。
欢迎拉动要求!!
先决条件
虚拟环境
PyTorch(经测试为0.3.0,但可以使用0.4.0)
准备数据集
下载LOOKBOOK数据集:
运行sh setup.sh
训练
python main.py \
--gpu_id 0 \
--root_dir 'data/prepro' \
--csv_file 'data/prepro/label.csv' \
--expr 'experiment1' \
--batch_size 24 \
--load_size 64 \
--lr 0.0002
可视化
tensorboard --logdir repo/
【文件预览】:
pixel-dt-gan-pytorch-master
----config.py(2KB)
----dataloader.py(3KB)
----tf.py(4KB)
----utils.py(4KB)
----main.py(2KB)
----requirements.txt(410B)
----pixel_dt_gan.py(6KB)
----README.md(1KB)
----layers.py(938B)
----data()
--------prepro()
----.gitignore(120B)
----setup.sh(181B)
----solver.py(8KB)
----prepro.py(1KB)