celebA-cyclegan:在celebA数据集上训练的cycle-gan的实现

时间:2024-06-14 17:59:06
【文件属性】:

文件名称:celebA-cyclegan:在celebA数据集上训练的cycle-gan的实现

文件大小:1.93MB

文件格式:ZIP

更新时间:2024-06-14 17:59:06

Python

在上的cycle-gan的实现。 使用带有梯度罚分的wgan来稳定训练并避免模型崩溃。 运行代码 安装要求: git clone https://github.com/MorvanZhou/celebA-cyclegan cd celebA-cyclegan pip3 install -r requirements.txt 从下载CelebA img_align_celeba.zip(〜1.4GB) 和list_attr_celeba.txt(25MB)从 。 解析数据 python dataset.py --data_dir ~ /data/celebA_img_align/ 训练 python train.py --data_dir ~ /data/celebA_img_align/ --soft_gpu -b 32 --epoch 51 --cycle_lambda 10 -


【文件预览】:
celebA-cyclegan-main
----cyclegan.py(5KB)
----utils.py(2KB)
----restore.py(3KB)
----train.py(4KB)
----dataset.py(5KB)
----cnn.py(4KB)
----anaylize.py(906B)
----requirements.txt(80B)
----.gitignore(39B)
----demo()
--------ep049t1000.png(1.92MB)
----README.md(1KB)

网友评论