文章目录
前言(Preface)
此博客记录三种基于深度学习的图像去雾网络,简述其算法特点并定性定量分析网络性能。
GFN-dehazing
原文:Gated Fusion Network for Single Image Dehazing(用于单幅图像去雾的门控融合网络)
原文链接: Gated Fusion Network for Single Image Dehazing
原文开源代码:GFN-dehazing
Dehaze-cGAN
原文:Single Image Dehazing via Conditional Generative Adversarial Network(基于条件生成对抗网络的单张图像去雾算法)
原文链接: Single Image Dehazing via Conditional Generative Adversarial Network
原文开源代码:Dehaze-cGAN
Cycle-Dehaze
原文:Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing(Cycle-Dehaze: 用于单幅图像去雾的增强型 CycleGAN 网络)
原文链接: Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing
原文开源代码:Cycle-Dehaze
一、GFN-dehazing
由于大气光照的影响,雾霾图像中的颜色通常发生变化,并且光照散射和颜色衰减的影响,使图像在远处缺乏可见性。基于这两点原因,GFN-dehazing借助三类特殊类型的图像,完成图像去雾。
白平衡图像:白平衡图像作为输入的目的是消除大气颜色造成的色差。白平衡图像可以消除大气光引起的颜色偏移,其结果呈现低对比度,通过加入另外两项输入来提高对比度。
对比度图像:通过在模糊输入中减去整张图像的平均亮度值导出介质图像,然后使用带有亮度信息的参数线性恢复介质图像中的亮度,最终得到对比度增强图像;对比度增强图像会导致图像中较暗区域趋向于黑色。
伽马校正图像:使用伽马校正图像来弥补对比度增强图像中趋于黑色所丢失的信息。
下图分别为原始图像、白平衡图像、对比度图像和伽马校正图像。
下图为GFN-dehazing网络的整体结构。网络输入原始的雾霾图像及三组派生图像(白平衡图像、对比度图像、伽马校正图像),在末端处输出对应三组派生图像的权重,然后和原始的三组图像分别进行加权融合,合成去雾图像。
直接将网络输出的权重置信图和派生图像进行加权相加可能会造成输出的去雾图像产生伪影。所以通过改变图像的分辨率使用多尺度细化的方法提高输出图片的质量。下图为我根据文章思路对多尺度细化的理解,其中每个尺度下输入图片的分辨率不同(图片的大小不同),此处所使用的网络均为同一个网络,所用的损失函数不同。
第k尺度的损失函数为 L c o n t [ θ , k ] = 1 N ∑ ∣ ∣ F ( I i , k , θ , k ) − J i , k ∣ ∣ L_{cont}[θ, k] = \frac{1}{N}\sum ||F(I_{i,k}, \theta, k) - J_{i,k}|| Lcont[θ,k]=N1∑∣∣F(I