【Pytorch】【PULSE】基于StyleGan的高分辨率图像生成器

因为研究项目是高分辨率图像生成器，所以我们还是从github学习已有的代码（主要还是从别的博主的博客借鉴了很多！）应该所有人都有体会，就是别人能运行的代码！我绝对要疯狂Debug。希望这次能快点成功噢。

学习资料

开源代码：https://github.com/adamian98/pulse

PULSE + ALAE：使用 StyleGAN 从低阶图像重建高清人脸

心中无码便是高清

环境配置报错参考

我的环境配置：

• 显卡NAVIDIA 1050Ti，版本411.31
• cuda 10.0
• cudann
• 配置路径Pytorch
  

cuda安装教程+cudnn安装教程
Windows10检查Cuda和cuDNN是否安装成功
windows10 Anaconda安装pytorch(GPU)

解决torch.cuda.is_available()一直返回False的玄学方法之一

这真的是唯一有效的解决方法，我下载的是（随意挑的）

报错：AttributeError: ‘Tensor’ object has no attribute ‘ndim’

就是把所有的param.ndim改成len(list(param.size()))

成功运行PULSE开源项目时的环境

运行步骤

1. 将测试图片（高清）放在工程realpics文件夹内
2. 使用 align_face.py 对图片降低分辨率处理，dlib检测人脸框默认降低为 32x32 分辨率的图片。python align_face.py
   降阶后的图片保存到：./input 目录中
3. 运行：run.py基于 32x32 的人脸小图，重建生成 1024x1024 的高清人脸大图，放在：./runs 目录中。
   或者可以python run.py -steps=2000指定迭代次数

第一次的尝试

有点拉胯。。。

原图

处理后低分辨率图像

图像还原结果