GAN深度学习·神经网络·计算机视觉深度学习-推荐系统-CV-NLP

SRGAN阅读笔记

2017-05-29  本文已影响1137人  Mordekaiser

论文:Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial
Github:SRGAN

Abstract

CNN在传统的单帧超分辨率重建上取得了非常好的效果,可以取得较高的峰值信噪比(PSNR)。他们大都以MSE为最小化的目标函数,这样虽然可以取得较高的峰值信噪比,但是当图像下采样倍数较高时,重建的得到的图片会过于平滑,丢失细节。本文提出了利用GAN来进行超分辨率重建的方法,该网络是第一个能恢复4倍下采样图像的框架。作者提出的损失函数有两部分组成:对抗损失,内容损失。
对抗损失将图像映射到高位流形空间,并用判别网络去判别重建后的图像和原始图像。而内容损失则是基于感觉相似性(perceptual similarity)而非像素相似性(pixel similarity)。

Intruction

所谓超分辨率重建就是将低分辨率图像恢复成对应的高分辨率图像。但是由于低分辨率图像信息的缺失,这是一个病态的求逆问题,尤其是在恢复的倍数较高的时候。传统的方法考虑加入一些先验信息以恢复高分辨率图像(如插值法),此外,还有基于回归方法的随记森林也可以用于超分问题。稀疏学习,CNN在超分上也取得了非常好的效果,特别是CNN,使得超分已经可以达到实时应用。

Method

作者提出的网络结构如下:

网络结构

生成网络由残差结构组成,并结合BN,生成的图像丢到判别网络中,交给判别器判断是生成的高分辨率图像还是真正的高分辨率图像。

作者主要的创新点在于代价函数的设计,尤其是将逐像素损失替换为内容损失。作者提出的损失函数由以下三部分加权组成:

正则损失

Experiments

作者用sub-pixel网络作为生成网络,用VGG作为判别网络构建GAN得到了非常好的结果,但是这个用的是逐像素差作为损失函数。

之后,作者尝试了自己提出的感知损失函数作为优化目标,虽然PSNR和SSIM不高,但是视觉效果都要优于其他网络,避免了其他方法的过度平滑的特性。

Conclusion

作者本文的贡献主要有两点:

上一篇 下一篇

猜你喜欢

热点阅读