MultiScaleConvNets





1. 概述

网络结构如图1所示。作者训练一个融合多尺度特征的网络来预测透射图,通过选择传输图t(x)中0.1%最暗像素来估计大气光 A 。在这些像素中,选取对应的朦胧图像 I 中强度最高的像素作为大气光。

网络输出的透射图尺寸与GT相同,去掉上、下采样虽能达到同样效果但是损害了模型性能。作者通过将前一层的一个像素点赋予后一层4个像素点同样的值来实现上采样。




在这里插入图片描述

图1 网络结构

作者修改了原始大气散射模型公式来进行预测:




在这里插入图片描述

2. 损失函数

损失函数如下:




在这里插入图片描述

作者利用数据集提供的景深来创建透射图,并用将其与模型预测的透射图计算MSE。

3. 数据集

  1. NYU Depth dataset
  2. Middlebury stereo database

这两个数据集都提供了景深,作者分别生成了18000张雾图和透射图作训练集,所有的训练图像都被调整为320×240像素的标准尺寸;50张雾图作验证集。

4. 参考文献

[1] Wenqi Ren, Si Liu, Hua Zhang, Jinshan Pan, Xiaochun Cao1(B),
and Ming-Hsuan Yang3. “Single Image Dehazing via Multi-scale Convolutional Neural Networks”. Springer International Publishing AG 2016, B. Leibe et al. (Eds.): ECCV 2016, Part II, LNCS 9906, pp. 154–169, 2016.DOI: 10.1007/978-3-319-46475-6 10.

发布了119 篇原创文章 ·
获赞 8 ·
访问量 6919