文件名称:ConSinGAN:PyTorch实施的“用于训练单图像GAN的改进技术”(WACV-21)
文件大小:71.64MB
文件格式:ZIP
更新时间:2024-04-09 02:06:46
image image-editing gan image-generation harmonization
康辛甘 正式发表了论文 。 有关本文的简短摘要,请参见我们的。 附加数据:和。 我们研究并推荐了在单个图像上训练GAN的新技术。我们的模型是在原始训练图像的几种不同分辨率上进行迭代训练的,其中分辨率随着训练的进行而增加。每当我们增加训练图像的分辨率时,我们都会通过添加其他卷积层来增加生成器的容量。在给定的时间,我们不会训练完整的模型,而只会训练其中的一部分,即最近添加的卷积层。最新的卷积层以给定的学习率进行训练,而先前存在的卷积层以较小的学习率进行训练。 安装 python 3.5 pytorch 1.1.0 pip install -r requirements.txt 无条件生成 要使用本文运行中的默认参数训练模型,请执行以下操作: python main_train.py --gpu 0 --train_mode generation --input_name Images/G