Nothing Special   »   [go: up one dir, main page]

skip to main content
research-article

A deep Retinex network for underwater low-light image enhancement

Published: 16 October 2023 Publication History

Abstract

Underwater images suffer from color cast and low contrast due to the light absorption and scattering. Especially when natural light is not sufficient, large dark areas appear in the captured image, making it impossible to understand the image content. To address this issue, we propose an underwater low-light enhancement method based on Retinex theory. Our model is an end-to-end trainable. The decomposition network decomposes the raw image into reflectance and illumination according to Retinex theory. In the reflectance enhancement network, cross-residual blocks and dense connections can improve the efficiency of feature utilization and the hybrid attention concentrate on the regions of interest in feature maps from different perspectives. The illumination adjustment network utilizes adaptive frequency convolutional blocks to generate additional band information, which reconstructs the more natural illumination. In order to preserve the color consistency of the enhanced image with the reference image, we project the HSV space into the Cartesian coordinate system and use the Euclidean distance as the color cast loss to constrain the enhancement network. Qualitative and quantitative evaluations on different underwater datasets indicate that our method has the excellent performance and can achieve delightful visual enhancements.

References

[1]
Rahman Z, Aamir M, Ali Z, Saudagar AKJ, AlTameem A, and Muhammad K Efficient contrast adjustment and fusion method for underexposed images in industrial cyber-physical systems IEEE Syst. J. 2023
[2]
Rahman Z, Ali Z, Khan I, Uddin MI, Guan Y, and Hu Z Diverse image enhancer for complex underexposed image J. Electron. Imaging 2022 31 4 041213-041213
[3]
Guo X, Li Y, and Ling H Lime: Low-light image enhancement via illumination map estimation IEEE Trans. Image Process. 2016 26 2 982-993
[4]
Rahman Z, Yi-Fei P, Aamir M, Wali S, and Guan Y Efficient image enhancement model for correcting uneven illumination images IEEE Access 2020 8 109038-109053
[5]
Shi Z, Zhu MM, Guo B, Zhao M, and Zhang C Nighttime low illumination image enhancement with single image using bright/dark channel prior EURASIP J. Image Video Process. 2018 2018 1-15
[6]
He K, Sun J, and Tang X Single image haze removal using dark channel prior IEEE Trans. Pattern Anal. Mach. Intell. 2010 33 12 2341-2353
[7]
Zhang, W., Li, G., Ying, Z.: A new underwater image enhancing method via color correction and illumination adjustment. In: IEEE Visual Communications and Image Processing (VCIP) (2017), pp. 1–4. IEEE (2017)
[8]
Zhou J, Zhang D, and Zhang W Multiscale fusion method for the enhancement of low-light underwater images Math. Probl. Eng. 2020 2020 1-15
[9]
Liu X, Ma W, Ma X, and Wang J LAE-Net: a locally-adaptive embedding network for low-light image enhancement Pattern Recogn. 2023 133
[10]
Li C, Anwar S, Hou J, Cong R, Guo C, and Ren W Underwater image enhancement via medium transmission-guided multi-color space embedding IEEE Trans. Image Process. 2021 30 4985-5000
[11]
Woo, S., Park, J., Lee, J.-Y., Kweon, I.S.: Cbam: convolutional block attention module. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 3–19 (2018)
[12]
Li C, Anwar S, and Porikli F Underwater scene prior inspired deep underwater image and video enhancement Pattern Recogn. 2020 98
[13]
Wei, C., Wang, W., Yang, W., Liu, J.: Deep retinex decomposition for low-light enhancement. arXiv preprint arXiv:1808.04560 (2018)
[14]
Li C, Guo C, Ren W, Cong R, Hou J, Kwong S, and Tao D An underwater image enhancement benchmark dataset and beyond IEEE Trans. Image Process. 2019 29 4376-4389
[15]
Islam MJ, Xia Y, and Sattar J Fast underwater image enhancement for improved visual perception IEEE Robot. Autom. Lett. 2020 5 2 3227-3234
[16]
Hitam, M.S., Awalludin, E.A., Yussof, W.N.J.H.W., Bachok, Z.: Mixture contrast limited adaptive histogram equalization for underwater image enhancement. In: 2013 International Conference on Computer Applications Technology (ICCAT), pp. 1–5. IEEE (2013)
[17]
Singh K, Kapoor R, and Sinha SK Enhancement of low exposure images via recursive histogram equalization algorithms Optik 2015 126 20 2619-2625
[18]
Bai L, Zhang W, Pan X, and Zhao C Underwater image enhancement based on global and local equalization of histogram and dual-image multi-scale fusion IEEE Access 2020 8 128973-128990
[19]
Zhou J, Wei X, Shi J, Chu W, and Zhang W Underwater image enhancement method with light scattering characteristics Comput. Electr. Eng. 2022 100
[20]
Fu, X., Zhuang, P., Huang, Y., Liao, Y., Zhang, X.-P., Ding, X.: A retinex-based enhancing approach for single underwater image. In: 2014 IEEE International Conference on Image Processing (ICIP), pp. 4572–4576. IEEE (2014)
[21]
Rahman Z, Bhutto JA, Aamir M, Dayo ZA, and Guan Y Exploring a radically new exponential retinex model for multi-task environments J. King Saud Univ. Comput. Inf. Sci. 2023 35 7
[22]
Zhuang P and Ding X Underwater image enhancement using an edge-preserving filtering retinex algorithm Multimedia Tools Appl. 2020 79 17257-17277
[23]
Ghani ASA Image contrast enhancement using an integration of recursive-overlapped contrast limited adaptive histogram specification and dual-image wavelet fusion for the high visibility of deep underwater image Ocean Eng. 2018 162 224-238
[24]
Hou, G.-J., Luan, X., Song, D.-L.: A study on color model selection for underwater color image preprocessing. In: 2015 IEEE International Conference on Cyber Technology in Automation, Control, and Intelligent Systems (CYBER), pp. 1456–1461. IEEE (2015)
[25]
Rahman Z, Pu Y-F, Aamir M, and Wali S Structure revealing of low-light images using wavelet transform based on fractional-order denoising and multiscale decomposition Vis. Comput. 2021 37 5 865-880
[26]
Drews, P., Nascimento, E., Moraes, F., Botelho, S., Campos, M.: Transmission estimation in underwater single images. In: Proceedings of the IEEE International Conference on Computer Vision Workshops, pp. 825–830 (2013)
[27]
Drews PL, Nascimento ER, Botelho SS, and Campos MFM Underwater depth estimation and image restoration based on single images IEEE Comput. Graph. Appl. 2016 36 2 24-35
[28]
Yu H, Li X, Lou Q, Lei C, and Liu Z Underwater image enhancement based on DCP and depth transmission map Multimedia Tools Appl. 2020 79 20373-20390
[29]
Berman D, Levy D, Avidan S, and Treibitz T Underwater single image color restoration using haze-lines and a new quantitative dataset IEEE Trans. Pattern Anal. Mach. Intell. 2020 43 8 2822-2837
[30]
Wang Y, Liu H, and Chau L-P Single underwater image restoration using adaptive attenuation-curve prior IEEE Trans. Circuits Syst. I Regul. Pap. 2017 65 3 992-1002
[31]
Wang Y, Guo J, Gao H, and Yue H Uiec^ 2-net: CNN-based underwater image enhancement using two color space Signal Process.: Image Commun. 2021 96
[32]
Wu S, Luo T, Jiang G, Yu M, Xu H, Zhu Z, and Song Y A two-stage underwater enhancement network based on structure decomposition and characteristics of underwater imaging IEEE J. Ocean. Eng. 2021 46 4 1213-1227
[33]
Liu S, Fan H, Lin S, Wang Q, Ding N, and Tang Y Adaptive learning attention network for underwater image enhancement IEEE Robot. Autom. Lett. 2022 7 2 5326-5333
[34]
Li J, Skinner KA, Eustice RM, and Johnson-Roberson M WaterGAN: unsupervised generative network to enable real-time color correction of monocular underwater images IEEE Robot. Autom. Lett. 2017 3 1 387-394
[35]
Fabbri, C., Islam, M.J., Sattar, J.: Enhancing underwater imagery using generative adversarial networks. In: 2018 IEEE International Conference on Robotics and Automation (ICRA), pp. 7159–7165. IEEE (2018)
[36]
Zong X, Chen Z, and Wang D Local-CycleGAN: a general end-to-end network for visual enhancement in complex deep-water environment Appl. Intell. 2021 51 1947-1958
[37]
Peng, L., Zhu, C., Bian, L.: U-shape transformer for underwater image enhancement. In: Computer Vision–ECCV 2022 Workshops: Tel Aviv, Israel, October 23–27, 2022, Proceedings, Part II, pp. 290–307. Springer (2023)
[38]
Porto Marques T, Branzan Albu A, and Hoeberechts M A contrast-guided approach for the enhancement of low-lighting underwater images J. Imaging 2019 5 10 79
[39]
Marques, T.P., Albu, A.B.: L2uwe: A framework for the efficient enhancement of low-light underwater images using local contrast and multi-scale fusion. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, pp. 538–539 (2020)
[40]
Zhao, W., Rong, S., Ma, J., He, B.: Nonuniform illumination correction for underwater images through a pseudo-Siamese network. In: 2022 26th International Conference on Pattern Recognition (ICPR), pp. 1329–1335. IEEE (2022)
[41]
Ji, T.-L., Sundareshan, M.K., Roehrig, H.: Adaptive image contrast enhancement based on human visual properties. IEEE Trans. Med. Imaging 13(4), 573–586 (1994).
[42]
Panetta, Karen A., Wharton, Eric J., Agaian, Sos S.: Human visual system-based image enhancement and logarithmic contrast measure. IEEE Trans. Syst. Man Cybern. Part B (Cybernetics) 38(1), 174–188 (2008).
[43]
Luo, M.R., Cui, G., Rigg, B.: The development of the CIE 2000 colour-difference formula: Ciede2000. Color Research & Application: Endorsed by Inter-Society Color Council, The Colour Group (Great Britain), Canadian Society for Color, Color Science Association of Japan, Dutch Society for the Study of Color, The Swedish Colour Centre Foundation, Colour Society of Australia, Centre Français de la Couleur 26(5), 340–350 (2001)
[44]
Lv, F., Liu, B., Lu, F.: Fast enhancement for non-uniform illumination images using light-weight CNNs. In: Proceedings of the 28th ACM International Conference on Multimedia, pp. 1450–1458 (2020)
[45]
Lin S, Tang F, Dong W, Pan X, and Xu C SMNet: synchronous multi-scale low light enhancement network with local and global concern IEEE Trans. Multimedia 2023
[46]
Johnson, J., Alahi, A., Fei-Fei, L.: Perceptual losses for real-time style transfer and super-resolution. In: Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands, October 11–14, 2016, Proceedings, Part II 14, pp. 694–711. Springer, Berlin (2016)
[47]
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556 (2014)
[48]
Deng, J., Dong, W., Socher, R., Li, L.-J., Li, K., Fei-Fei, L.: Imagenet: a large-scale hierarchical image database. In: 2009 IEEE Conference on Computer Vision and Pattern Recognition, pp. 248–255. IEEE (2009)
[49]
Ancuti, C., Ancuti, C.O., Haber, T., Bekaert, P.: Enhancing underwater images and videos by fusion. In: 2012 IEEE Conference on Computer Vision and Pattern Recognition, pp. 81–88. IEEE (2012)
[50]
Panetta K, Gao C, and Agaian S Human-visual-system-inspired underwater image quality measures IEEE J. Ocean. Eng. 2015 41 3 541-551
[51]
Mittal A, Soundararajan R, and Bovik AC Making a “completely blind” image quality analyzer IEEE Signal Process. Lett. 2012 20 3 209-212
[52]
Wang S, Ma K, Yeganeh H, Wang Z, and Lin W A patch-structure representation method for quality assessment of contrast changed images IEEE Signal Process. Lett. 2015 22 12 2387-2390

Cited By

View all
  • (2024)Thermal infrared action recognition with two-stream shift Graph Convolutional NetworkMachine Vision and Applications10.1007/s00138-024-01550-235:4Online publication date: 13-May-2024
  • (2024)A tree-based approach for visible and thermal sensor fusion in winter autonomous drivingMachine Vision and Applications10.1007/s00138-024-01546-y35:4Online publication date: 3-May-2024

Recommendations

Comments

Please enable JavaScript to view thecomments powered by Disqus.

Information & Contributors

Information

Published In

cover image Machine Vision and Applications
Machine Vision and Applications  Volume 34, Issue 6
Nov 2023
561 pages

Publisher

Springer-Verlag

Berlin, Heidelberg

Publication History

Published: 16 October 2023
Accepted: 25 September 2023
Revision received: 20 August 2023
Received: 09 May 2023

Author Tags

  1. Underwater images
  2. Low-light enhancement
  3. Retinex
  4. Deep learning

Qualifiers

  • Research-article

Funding Sources

Contributors

Other Metrics

Bibliometrics & Citations

Bibliometrics

Article Metrics

  • Downloads (Last 12 months)0
  • Downloads (Last 6 weeks)0
Reflects downloads up to 11 Jan 2025

Other Metrics

Citations

Cited By

View all
  • (2024)Thermal infrared action recognition with two-stream shift Graph Convolutional NetworkMachine Vision and Applications10.1007/s00138-024-01550-235:4Online publication date: 13-May-2024
  • (2024)A tree-based approach for visible and thermal sensor fusion in winter autonomous drivingMachine Vision and Applications10.1007/s00138-024-01546-y35:4Online publication date: 3-May-2024

View Options

View options

Media

Figures

Other

Tables

Share

Share

Share this Publication link

Share on social media