PD-Net:基于金字塔池化模块改进的Dehaze-Net
PD-Net: Improved Dehaze-Net Based on Pyramid Pooling Module
DOI: 10.12677/AAM.2021.1010352, PDF,    国家自然科学基金支持
作者: 宋翼洋, 刘 刚:成都信息工程大学应用数学学院,四川 成都
关键词: 去雾Dehaze-Net金字塔池化模块PD-NetDefogging Dehaze-Net Pyramid Pooling Module PD-Net
摘要: 对单幅雾天图像进行去雾是一个困难的任务,本文提出了一种名叫PD-Net的神经网络模型用于单幅图像去雾。该模型在Dehaze-Net模型中引入了金字塔池化模块且额外添加残差块。金字塔池化模块增强了模型对全局信息的提取,残差块有效地抑制了梯度消失现象。在仿真实验中,本文使用RESIDE数据集中的室内合成有雾图片作为实验数据集,定量分析比较了各种去雾方法对实验数据集的去雾结果,PD-Net模型展现了良好的性能。在真实实验中,对RESIDE数据集中的室外真实有雾图片进行去雾分析。真实实验的结果表明,相较于其他算法,PD-Net模型在大面积的天空区域及图片的细节处有更好的效果。
Abstract: It is a difficult task to remove haze from a single input image. In this paper, we present a neural network model called PD-Net for dehazing single image. In this model, Pyramid Pooling Module is introduced into Dehaze-Net model, and Residual blocks are added. The pyramid pooling module enhances the extraction of global information, and the residual block effectively suppresses the disappearance of gradient. In the simulation experiment, we use the indoor synthetic foggy pictures in the RESIDE set as the experimental set, quantitatively analyze and compare the defogging results of various defogging methods on the experimental set, and the PD-Net model shows good performance. In the real experiment, we defog and analyze the images which is the outdoor real foggy pictures in the RESIDE set. The results of real experiments show that compared with other algorithms, PD-Net model has better effect in large-area sky area and details of picture.
文章引用:宋翼洋, 刘刚. PD-Net:基于金字塔池化模块改进的Dehaze-Net[J]. 应用数学进展, 2021, 10(10): 3351-3360. https://doi.org/10.12677/AAM.2021.1010352

参考文献

[1] izer, S.M., Amburn, E.P., Austin, J.D., et al. (1987) Adaptive Histogram Equalization and Its Variations. Computer Vision Graphics & Image Processing, 39, 355-368. [Google Scholar] [CrossRef
[2] Adelmann, H.G. (1998) Butterworth Equations for Homomorphic Filtering of Images. Computers in Biology & Medicine, 28, 169-181. [Google Scholar] [CrossRef
[3] Finlayson, G.D., Hordley, S.D. and Drew, M.S. (2002) Removing Shadows from Images Using Retinex. Proceedings Color Imaging Conference, Scottsdale, 12-15 November 2002, 73-79. [Google Scholar] [CrossRef
[4] Narasimhan, S.G. and Nayar, S.K. (2003) Contrast Restoration of Weather Degraded Images. IEEE Transactions on Pattern Analysis & Machine Intelligence, 25, 713-724. [Google Scholar] [CrossRef
[5] He, K., Jian, S., Tang, X.O., et al. (2011) Single Image Haze Removal Using Dark Channel Prior. IEEE Transactions on Pattern Analysis & Machine Intelligence, 33, 2341-2353. [Google Scholar] [CrossRef
[6] Cai, B., Xu, X., Jia, K., et al. (2016) Dehaze-Net: An End-to-End System for Single Image Haze Removal. IEEE Transactions on Image Processing, 25, 5187-5198. [Google Scholar] [CrossRef
[7] Ren, W., Si, L., Hua, Z., et al. (2016) Single Image Dehazing via Multi-Scale Convolutional Neural Networks. In: European Conference on Computer Vision, Springer, Cham, 154-169. [Google Scholar] [CrossRef
[8] Li, B., Peng, X., Wang, Z., et al. (2017) AOD-Net: All-in-One Dehazing Network. 2017 IEEE International Conference on Computer Vision (ICCV), Venice, 22-29 October 2017, 4780-4788. [Google Scholar] [CrossRef
[9] Zhao, H., Shi, J., Qi, X., et al. (2017) Pyramid Scene Parsing Network. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, 21-26 July 2017, 6230-6239. [Google Scholar] [CrossRef
[10] Szegedy, C., Wei, L., Jia, Y., et al. (2015) Going Deeper with Convolutions. 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Boston, 7-12 June 2015, 1-9. [Google Scholar] [CrossRef
[11] Lecun, Y. and Bottou, L. (1998) Gradient-Based Learning Applied to Document Recognition. Proceedings of the IEEE, 86, 2278-2324. [Google Scholar] [CrossRef
[12] He, K., Zhang, X., Ren, S., et al. (2016) Deep Residual Learning for Image Recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 770-778. [Google Scholar] [CrossRef
[13] Simonyan, K. and Zisserman, A. (2014) Very Deep Convolutional Networks for Large-Scale Image Recognition.
[14] Zuiderveld, K. (1994) Contrast Limited Adaptive Histogram Equalization. In: Graphics Gems IV, Morgan Kaufmann, Burlington, 474-485. [Google Scholar] [CrossRef
[15] Wang, Z., Bovik, A.C., Sheikh, H.R., et al. (2004) Image Quality Assessment: From Error Visibility to Structural Similarity. IEEE Transactions on Image Processing, 13, 600-612. [Google Scholar] [CrossRef