Deep Learning-Driven Parameter Adaptation for Underwater Image Restoration
DOI:
https://doi.org/10.5753/reic.2024.4671Abstract
In this paper we propose a learning-based approach to enhance underwater image quality by optimizing parameters and applying intensity transformations. Our methodology involves training a CNN Regression model on diverse underwater images to learn enhancing parameters, followed by applying intensity transformation techniques. In order to evaluate our approach, we conducted experiments using well-known underwater image datasets found in the literature, comprising real-world subaquatic images and we propose a novel underwater image dataset, composed by 276 images from Amazon turbid water rivers. The results demonstrate that our approach achieves an impressive accuracy rate in three different underwater image datasets. This high level of accuracy showcases the robustness and efficiency of our proposed method in restoring underwater images.
Downloads
References
Ancuti, C. O., Ancuti, C., De Vleeschouwer, C., and Bekaert, P. (2018). Color balance and fusion for underwater image enhancement. IEEE Trans. on Image Processing, 27(1):379–393.
Drews, P. L., Nascimento, E. R., Botelho, S. S., and Montenegro Campos, M. F. (2016). Underwater depth estimation and image restoration based on single images. IEEE Computer Graphics and Applications, 36(2):24–35.
Fu, X., Zhuang, P., Huang, Y., Liao, Y., Zhang, X.-P., and Ding, X. (2014). A retinex-based enhancing approach for single underwater image. In 2014 IEEE International Conference on Image Processing (ICIP), pages 4572–4576.
Gangisetty, S. and Rai, R. R. (2022). Floodnet: Underwater image restoration based on residual dense learning. Signal Processing: Image Communication, 104:116647.
Huang, Y., Yuan, F., Xiao, F., and Cheng, E. (2022). Underwater image enhancement based on color restoration and dual image wavelet fusion. Signal Processing: Image Communication, 107:116797.
Jiang, Qiuping Li, F. L. C. L. D. D. J. A. (2022). Underwater imaging formation model-embedded multiscale deep neural network for underwater image enhancement. Mathematical Problems in Engineering, 13(4).
Kai, Y., Lanyue, L., Ziqiang, Z., Guoqing, W., and Yang, Y. (2022). Medium transmission map matters for learning to restore real-world underwater images.
Komal, V. and Yaduvir, S. (2011). Enhancement of images using histogram processing techniques. International Journal of Computer Technology and Applications, 02.
Li, C., Anwar, S., and Porikli, F. (2020a). Underwater scene prior inspired deep underwater image and video enhancement. Pattern Recognition, 98:107038.
Li, C., Guo, C., Ren, W., Cong, R., Hou, J., Kwong, S., and Tao, D. (2020b). An underwater image enhancement benchmark dataset and beyond. IEEE Transactions on Image Processing, 29:4376–4389.
Li, C., Tang, S., Kwan, H. K., Yan, J., and Zhou, T. (2020c). Color correction based on cfa and enhancement based on retinex with dense pixels for underwater images. IEEE Access, 8:155732–155741.
Liu, S., Fan, H., Lin, S., Wang, Q., Ding, N., and Tang, Y. (2022a). Adaptive learning attention network for underwater image enhancement. IEEE Robotics and Automation Letters, 7(2):5326–5333.
Liu, Y., Xu, H., Zhang, B., Sun, K., Yang, J., Li, B., Li, C., and Quan, X. (2022b). Model-based underwater image simulation and learning-based underwater image enhancement method. Information, 13(4).
Lyu, Z., Peng, A., Wang, Q., and Ding, D. (2022). An efficient learning-based method for underwater image enhancement. Displays, 74:102174.
Martinho, L., Neto, O., Calvalcanti, J., Pio, J., and Oliveira, F. (2023). An approach for fish detection in underwater images. In Anais do XVIII Workshop de Visão Computacional, pages 6–11, Porto Alegre, RS, Brasil. SBC.
Martinho, L. A., Calvalcanti, J. a. M. B., Pio, J. L., and Oliveira, F. G. (2024). Diving into clarity: Restoring underwater images using deep learning. J. Intell. Robotics Syst., 110(1).
Martinho, L. A., Oliveira, F. G., Cavalcanti, J. M. B., and Pio, J. L. S. (2022). Underwater image enhancement based on fusion of intensity transformation techniques. In 2022 Latin American Robotics Symposium (LARS), 2022 Brazilian Symposium on Robotics (SBR), and 2022 Workshop on Robotics in Education (WRE), pages 348–353.
Peng, Y.-T., Cao, K., and Cosman, P. C. (2018). Generalization of the dark channel prior for single image restoration. IEEE Trans. on Image Processing, 27(6):2856–2868.
Peng, Y.-T. and Cosman, P. C. (2017). Underwater image restoration based on image blurriness and light absorption. IEEE Trans. on Image Processing, 26(4):1579–1594.
Qiao, N., Dong, L., and Sun, C. (2023). Adaptive deep learning network with multi-scale and multi-dimensional features for underwater image enhancement. IEEE Transactions on Broadcasting, 69(2):482–494.
Sharma, P. K., Bisht, I., and Sur, A. (2022). Wavelength-based attributed deep neural network for underwater image restoration.
Tao, Y., Dong, L., and Xu, W. (2020). A novel two-step strategy based on white-balancing and fusion for underwater image enhancement. IEEE Access, 8:217651–217670.
Wang, Y., Zhao, Y., Pan, H., and Zhou, W. (2022). An improved reinforcement learning method for underwater image enhancement. In 2022 IEEE 25th Int. Conf. on Computer Supported Cooperative Work in Design (CSCWD), pages 1077–1082.
Zhao, W., Rong, S., Li, T., Feng, J., and He, B. (2023). Enhancing underwater imagery via latent low-rank decomposition and image fusion. IEEE Journal of Oceanic Engineering, 48(1):147–159.
Zheng, M. and Luo, W. (2022). Underwater image enhancement using improved cnn based defogging. Electronics, 11(1).
Zhuang, P., Li, C., and Wu, J. (2021). Bayesian retinex underwater image enhancement. Engineering Applications of Artificial Intelligence, 101:104171.
Zhuang, P., Wu, J., Porikli, F., and Li, C. (2022). Underwater image enhancement with hyper-laplacian reflectance priors. IEEE Trans. on Image Processing, 31:5442–5455.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 The authors
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
