文件名称:AnimeGAN-torch:使用Pytorch从AnimeGANv2复制结果
文件大小:28KB
文件格式:ZIP
更新时间:2024-03-07 09:09:49
Python
动漫火炬 此仓库是在Pytorch中复制AnimeGANv2的结果 这个分支是实现从NOGAN培训 。 由于GAN训练本身是不稳定的,因此希望尽可能少地进行GAN训练,同时仍然具有GAN的作用。 NOGAN在图像着色中效果很好,这是图像到图像翻译中的任务,从直观上讲,它也应适用于其他图像到图像翻译任务。 AnimeGAN的NOGAN培训的功能丧失使用相同的损失 。 NOGAN的培训在笔记本电脑上进行,该笔记本电脑需要大量培训,并且要经过很多检查点和反复试验。 该分支增加了FP16培训表格 。 请参阅设置为培训的不同级别的文档。 FP16培训不适用于GAN培训。
【文件预览】:
AnimeGAN-torch-master
----.gitignore(78B)
----init_train.py(1B)
----data_transform.py(488B)
----model()
--------__init__.py(76B)
--------discriminator.py(3KB)
--------layers.py(4KB)
--------__pycache__()
--------generator.py(2KB)
----trial.py(30KB)
----optimizers.py(855B)
----weights_init.py(259B)
----meter.py(2KB)
----optim()
--------omd.py(6KB)
--------__init__.py(48B)
--------extragradient.py(11KB)
----README.md(974B)
----test.py(2KB)
----edge_smooth.py(2KB)
----loss.py(5KB)
----dataset.py(3KB)
----easy_setup.sh(571B)
----train.py(4KB)