针对人脸识别中遮挡区域降低检测准确度的问题,提出一种基于生成对抗网络的遮挡人脸修复方法。该方法以生成对抗网络作为基本架构,结合Wasserstein距离和添加梯度惩罚损失函数来训练网络模型,以全局上下文损失和先验损失相结合的方式来约束网络生成无遮挡人脸图片,并利用泊松融合完成遮挡区域的修复。在CelebA数据集的实验结果表明,所提方法较其他文献模型训练更稳定,PSNR平均值提高了5%,SSIM平均值提高了8%。
1
WGAN-GP纯粹是为了满足WGAN中的李普西斯条件,WGAN自己的满足方式是gradient clipping,但是这样的话WGAN的大部分weight会是正负0.01,需要新的满足李普西斯条件的方法,这就是motivation;
2021-11-07 13:46:40 7KB WGAN-GP
1
GAN模型的Pytorch代码 这是使用相同的卷积架构的3种不同GAN模型的pytorch实现。 DCGAN(深度卷积GAN) WGAN-CP(使用重量修剪的Wasserstein GAN) WGAN-GP(使用梯度罚分的Wasserstein GAN) 依存关系 突出的软件包是: 麻木 scikit学习 张量流2.0 pytorch 1.6.0 火炬视觉0.7.0 要快速轻松地安装所有依赖项,您应该使用pip pip install - r requirements . txt 训练 在Fashion-MNIST数据集上运行DCGAN模型的训练: python main.py --model DCGAN \ --is_train True \ --download True \ -
2021-10-26 16:32:27 5.51MB Python
1
GAN的集合 无监督GAN的Pytorch实现。 有关使用pytorch计算初始分数和FID的更多信息,请参见 楷模 直流电源 威根 WGAN-GP SN-GAN 要求 初始化metric模型git submodule update --init 安装python包pip install -U pip setuptools pip install -r requirements.txt 结果 模型 数据集 起始分数 FID 直流电源 CIFAR10 6.04(0.05) 47.90 WGAN(CNN) CIFAR10 6.64(0.6) 33.27 WGAN-GP(CNN) CIFAR10 7.47(0.06) 24.00 WGAN-GP(ResNet) CIFAR10 7.74(0.10) 21.89 SNGAN(CNN) CIFAR10 7
2021-10-10 20:33:13 22KB pytorch dcgan wgan-gp sngan
1
建议 我们基于GAN的面部属性编辑工作 。 新闻 2019年4月8日:我们通过Tensorflow 2重新实现了这些GAN! 旧版本位于: 或“ v1”目录中。 GAN-Tensorflow 2 Tensorflow的2个实现 , , 和 。 样例结果 时尚MNIST DCGAN LSGAN WGAN-GP 德拉甘 名人 DCGAN LSGAN WGAN-GP 德拉甘 日本动漫 WGAN-GP 德拉甘 用法 环境 Python 3.6 TensorFlow 2.2,TensorFlow插件0.10.0 OpenCV,scikit映像,tqdm,oyaml 我们建议
2021-08-29 19:33:31 15.67MB tensorflow dcgan wgan lsgan
1
基于PyTorch实现生成对抗网络 拟合给定分布 要求可视化训练过程 实验报告 对比GAN、WGAN、WGAN-GP(稳定性、性能) 对比不同优化器的影响
2021-06-16 19:41:32 3.19MB wgan gan wgan-gp
1
CVAE-GAN-ZOOS-PyTorch-初学者 首先先感谢给小透明点赞的几个朋友。 中文讲解: 如果你是第一次接触AE自编码器和GAN生成对抗网络,那这将会是一个非常有用且效率的学习资源。所有的内容使用PyTorch编写,编写格式清晰,非常适合PyTorch新手作为学习资源。本项目的所有模型当前都是基于MNIST数据库进行图片生成。MNIST数据集是一个比较小,一个光CPU可以运行起来的小数据库。新人友好数据库。 本项目包含以下模型:AE(自编码器),DAE(降噪自编码器),VAE(变分自编码器),GAN(对抗生成网络),CGAN(条件对抗生成网络),DCGAN(深度)卷积对抗生成网络),WGAN(Wasserstain对抗生成网络),WGAN-GP(基于渐变惩罚的WGAN),VAE-GAN(变分自编码对抗生成网络),CVAE-GAN(条件变分自编码对抗生成)网络)PS:部分英文翻译的
2021-05-12 17:22:58 42.61MB Python
1
本人实现的基于pytorch的WGAN-GP超分辨率重建算法,可以正常训练和测试,训练集用的是Flickr2K和DIV2K,pytorch版本1.7.0,可视化需要tensorboard依赖库
2021-03-09 17:02:11 166.82MB pytorch 超分辨率重建 WGAN-GP 深度学习
1
基于PyTorch实现生成对抗网络 拟合给定分布 要求可视化训练过程 实验报告 对比GAN、WGAN、WGAN-GP(稳定性、性能) 对比不同优化器的影响
2019-12-21 20:29:29 254KB wgan gan gan-gp
1
基于tensorflow实现的wgan和wgan_gp,将数据集放于data文件夹下即可,如data/数据/*.jpg
2019-12-21 19:45:08 10KB WAGN WGAN_gp tensorflow
1