Abstract
Single-image dehazing is an ill-posed problem. Most previous methods focused on estimating intermediate parameters for input hazy images. In this paper, we propose a novel end-to-end Simplified Non-locally Dense Network (SNDN) which does not rely on intermediate parameters. To capture long-range dependencies, we propose a Simplified Non-local Dense Block (SNDB) which is lightweight and outperforms traditional non-local method. Our SNDB will be embedded into a densely connected encoder–decoder network. To avoid gradients vanishing problem, we propose a simple branch network which only have five convolution layers. The effectiveness of our proposed network is proved through ablation experiment. In addition, we enhanced our training set by synthesizing colored hazy images, which helps restore the original color of the hazy image. The experimental results demonstrate that our network have better performance than most of the pervious state-of-the-art methods.
Similar content being viewed by others
References
Ancuti, C., Ancuti, C.O., Timofte, R.: Ntire 2018 challenge on image dehazing: methods and results. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, pp. 891–901 (2018)
Ancuti, C.O., Ancuti, C., De Vleeschouwer, C., Sbetr, M.: Color channel transfer for image dehazing. IEEE Signal Process. Lett. 26(9), 1413–1417 (2019)
Berman, D., Treibitz, T., Avidan, S.: Single image dehazing using haze-lines. IEEE Trans. Pattern Anal. Mach. Intell. (2018)
Buades, A., Coll, B., Morel, J.M.: A non-local algorithm for image denoising. In: 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR’05), vol. 2. IEEE, pp. 60–65 (2005)
Cai, B., Xu, X., Jia, K., Qing, C., Tao, D.: Dehazenet: an end-to-end system for single image haze removal. IEEE Trans. Image Process. 25(11), 5187–5198 (2016)
Cao, Y., Xu, J., Lin, S., Wei, F., Hu, H.: Gcnet: Non-local networks meet squeeze-excitation networks and beyond. arXiv preprint arXiv:1904.11492 (2019)
Chen, D., He, M., Fan, Q., Liao, J., Zhang, L., Hou, D., Yuan, L., Hua, G.: Gated context aggregation network for image dehazing and deraining. In: 2019 IEEE Winter Conference on Applications of Computer Vision (WACV). IEEE, pp. 1375–1383 (2019)
Fattal, R.: Dehazing using color-lines. ACM Trans. Graph. 34(1), 1–14 (2014)
Gandelsman, Y., Shocher, A., Irani, M.: “double-dip”: Unsupervised image decomposition via coupled deep-image-priors. arXiv preprint arXiv:1812.00467 (2018)
Golts, A., Freedman, D., Elad, M.: Unsupervised single image dehazing using dark channel prior loss. IEEE Trans. Image Process. (2019)
He, K., Sun, J., Tang, X.: Single image haze removal using dark channel prior. IEEE Trans. Pattern Anal. Mach. Intell. 33(12), 2341–2353 (2010)
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. In: Proceedings of the IEEE Conference On Computer Vision and Pattern Recognition, pp. 7132–7141 (2018)
Huang, G., Liu, Z., Van Der Maaten, L., Weinberger, K.Q.: Densely connected convolutional networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 4700–4708 (2017)
Jégou, S., Drozdzal, M., Vazquez, D., Romero, A., Bengio, Y.: The one hundred layers tiramisu: fully convolutional densenets for semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, pp. 11–19 (2017)
Ju, M., Gu, Z., Zhang, D.: Single image haze removal based on the improved atmospheric scattering model. Neurocomputing 260, 180–191 (2017)
Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv preprint arXiv:1412.6980 (2014)
Li, B., Peng, X., Wang, Z., Xu, J.Z., Feng, D.: Aod-net: all-in-one dehazing network. In: Proceedings of the IEEE International Conference on Computer Vision (2017)
Li, B., Ren, W., Fu, D., Tao, D., Feng, D., Zeng, W., Wang, Z.: Benchmarking single-image dehazing and beyond. IEEE Trans. Image Process. 28(1), 492–505 (2018)
Li, G., He, X., Zhang, W., Chang, H., Dong, L., Lin, L.: Non-locally enhanced encoder-decoder network for single image de-raining. arXiv preprint arXiv:1808.01491 (2018)
Li, L., Dong, Y., Ren, W., Pan, J., Gao, C., Sang, N., Yang, M.H.: Semi-supervised image dehazing. IEEE Trans. Image Process. 29, 2766–2779 (2019)
McCartney, E.J.: Optics of the Atmosphere: Scattering by Molecules and Particles. Wiley, New York (1976)
Mei, K., Jiang, A., Li, J., Wang, M.: Progressive feature fusion network for realistic image dehazing. In: Asian Conference on Computer Vision. Springer, pp. 203–215 (2018)
Narasimhan, S.G., Nayar, S.K.: Chromatic framework for vision in bad weather. In: Proceedings IEEE Conference on Computer Vision and Pattern Recognition. CVPR 2000 (Cat. No. PR00662), vol. 1. IEEE, pp. 598–605 (2000)
Narasimhan, S.G., Nayar, S.K.: Vision and the atmosphere. Int. J. Comput. Vis. 48(3), 233–254 (2002)
Qu, Y., Chen, Y., Huang, J., Xie, Y.: Enhanced pix2pix dehazing network. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 8160–8168 (2019)
Ren, W., Liu, S., Zhang, H., Pan, J., Cao, X., Yang, M.H.: Single image dehazing via multi-scale convolutional neural networks. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) Computer Vision—ECCV 2016, pp. 154–169. Springer, Cham (2016)
Ren, W., Pan, J., Zhang, H., Cao, X., Yang, M.H.: Single image dehazing via multi-scale convolutional neural networks with holistic edges. Int. J. Comput. Vis. 2019, 1–20 (2019)
Sakaridis, C., Dai, D., Van Gool, L.: Semantic foggy scene understanding with synthetic data. Int. J. Comput. Vis. 126(9), 973–992 (2018)
Santra, S., Mondal, R., Chanda, B.: Learning a patch quality comparator for single image dehazing. IEEE Trans. Image Process. 27(9), 4598–4607 (2018)
Shi, L.F., Chen, B.H., Huang, S.C., Larin, A.O., Seredin, O.S., Kopylov, A.V., Kuo, S.Y.: Removing haze particles from single image via exponential inference with support vector data description. IEEE Trans. Multimed. 20(9), 2503–2512 (2018)
Sulami, M., Glatzer, I., Fattal, R., Werman, M.: Automatic recovery of the atmospheric light in hazy images. In: 2014 IEEE International Conference on Computational Photography (ICCP). IEEE, pp. 1–11 (2014)
Tan, R.T.: Visibility in bad weather from a single image. In: 2008 IEEE Conference on Computer Vision and Pattern Recognition. IEEE, pp. 1–8 (2008)
Tarel, J.P., Hautiere, N., Caraffa, L., Cord, A., Halmaoui, H., Gruyer, D.: Vision enhancement in homogeneous and heterogeneous fog. IEEE Intell. Transp. Syst. Mag. 4(2), 6–20 (2012)
Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, Ł., Polosukhin, I.: Attention is all you need. In: Advances in Neural Information Processing Systems, pp. 5998–6008 (2017)
Wang, X., Girshick, R., Gupta, A., He, K.: Non-local neural networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7794–7803 (2018)
Xiao, C., Gan, J.: Fast image dehazing using guided joint bilateral filter. Vis. Comput. 28(6–8), 713–721 (2012)
Yang, D., Sun, J.: Proximal dehaze-net: a prior learning-based deep network for single image dehazing. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 702–717 (2018)
Yang, Y., Lu, H.: Single image deraining using a recurrent multi-scale aggregation and enhancement network. In: 2019 IEEE International Conference on Multimedia and Expo (ICME). IEEE, pp. 1378–1383 (2019)
Zhang, H., Patel, V.M.: Densely connected pyramid dehazing network. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3194–3203 (2018)
Zhang, H., Sindagi, V., Patel, V.M.: Multi-scale single image dehazing using perceptual pyramid deep network. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, pp. 902–911 (2018)
Zhang, Y., Ding, L., Sharma, G.: Hazerd: an outdoor scene dataset and benchmark for single image dehazing. In: 2017 IEEE International Conference on Image Processing (ICIP). IEEE, pp. 3205–3209 (2017)
Zhao, D., Xu, L., Yan, Y., Chen, J., Duan, L.Y.: Multi-scale optimal fusion model for single image dehazing. Signal Process. Image Commun. 74, 253–265 (2019)
Zhu, Q., Mai, J., Shao, L.: A fast single image haze removal algorithm using color attenuation prior. IEEE Trans. Image Process. 24(11), 3522–3533 (2015)
Acknowledgements
This work was supported in part by the National Natural Science Foundation of China under Grant 61672228, Grant 61872241, Grant 61572316, and Grant 61370174, in part by the Shanghai Automotive Industry Science and Technology Development Foundation under Grant 1837, and in part by The Hong Kong Polytechnic University under Grant P0030419 and Grant P0030929.
Author information
Authors and Affiliations
Corresponding authors
Ethics declarations
Conflict of interest
All authors declare that they have no conflict of interest.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Chen, Z., Hu, Z., Sheng, B. et al. Simplified non-locally dense network for single-image dehazing. Vis Comput 36, 2189–2200 (2020). https://doi.org/10.1007/s00371-020-01929-y
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00371-020-01929-y