ABSTRACT
In the underwater environment, the color distortion and low contrast of the image are caused by environmental problems such as light absorption and scattering, which leads to the degradation of image quality. In order to improve the visual effect of the image, this paper proposed a color correction underwater image enhancement algorithm based on convolutional neural network. Firstly, a new underwater imaging model was used to correct the color cast problem of underwater images. Then, the convolutional neural network is used to extract the channel features of the corrected image, and the channel features are re-weighted by the multi-scale attention module to enhance the consistency of different feature maps, and the feature fusion is performed with the color corrected image. Finally, the image enhancement effect was improved by the fusion of features through the reconstruction calculation module. Experimental results show that the proposed algorithm can better correct the color distortion of the image and improve the image contrast. The main advantage is that the running speed of the proposed algorithm is two times faster than other advanced underwater image enhancement methods.
- At Drews P, Nascimento E, Moraes F, Transmission estimation in underwater single images[C]//Proceedings of the IEEE international conference on computer vision workshops. 2013: 825-830.Google Scholar
- Peng Y T, Cosman P C. Underwater image restoration based on image blurriness and light absorption[J]. IEEE transactions on image processing, 2017, 26(4): 1579-1594.Google ScholarDigital Library
- Bryson M, Johnson‐Roberson M, Pizarro O, True color correction of autonomous underwater vehicle imagery[J]. Journal of Field Robotics, 2016, 33(6): 853-874.Google ScholarDigital Library
- Iscar E, Barbalata C, Goumas N, Towards low cost, deep water AUV optical mapping[C]//OCEANS 2018 MTS/IEEE Charleston. IEEE, 2018: 1-6.Google Scholar
- Akkaynak D, Treibitz T. A revised underwater image formation model[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2018: 6723-6732.Google Scholar
- Solonenko M G, Mobley C D. Inherent optical properties of Jerlov water types[J]. Applied optics, 2015, 54(17): 5392-5401.Google Scholar
- Schettini R, Corchs S. Underwater image processing: state of the art of restoration and image enhancement methods[J]. EURASIP journal on advances in signal processing, 2010, 2010: 1-14.Google Scholar
- Narasimhan S G, Nayar S K. Contrast restoration of weather degraded images[J]. IEEE transactions on pattern analysis and machine intelligence, 2003, 25(6): 713-724.Google ScholarCross Ref
- He D M, Seet G G L. Divergent-beam lidar imaging in turbid water[J]. Optics and lasers in engineering, 2004, 41(1): 217-231.Google Scholar
- Schechner Y Y, Averbuch Y. Regularized image recovery in scattering media[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2007, 29(9): 1655-1660.Google ScholarDigital Library
- Li C Y, Guo J C, Cong R M, Underwater image enhancement by dehazing with minimum information loss and histogram distribution prior[J]. IEEE Transactions on Image Processing, 2016, 25(12): 5664-5677.Google ScholarDigital Library
- Drews P L J, Nascimento E R, Botelho S S C, Underwater depth estimation and image restoration based on single images[J]. IEEE computer graphics and applications, 2016, 36(2): 24-35.Google Scholar
- Gao S B, Zhang M, Zhao Q, Underwater image enhancement using adaptive retinal mechanisms[J]. IEEE Transactions on Image Processing, 2019, 28(11): 5580-5595.Google ScholarDigital Library
- Mur-Artal R, Montiel J M M, Tardos J D. ORB-SLAM: a versatile and accurate monocular SLAM system[J]. IEEE transactions on robotics, 2015, 31(5): 1147-1163.Google Scholar
- Chen X, Zhang P, Quan L, Underwater image enhancement based on deep learning and image formation model[J]. arXiv preprint arXiv:2101.00991, 2021.Google Scholar
- Roznere M, Li A Q. Real-time model-based image color correction for underwater robots[C]//2019 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS). IEEE, 2019: 7191-7196.Google Scholar
- Wang P, Chen P, Yuan Y, Understanding convolution for semantic segmentation[C]//2018 IEEE winter conference on applications of computer vision (WACV). Ieee, 2018: 1451-1460.Google Scholar
- Dai Y, Gieseke F, Oehmcke S, Attentional feature fusion[C]//Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision. 2021: 3560-3569.Google Scholar
- Underwater Object Detection Algorithm Contest[OL].http://uodac.pcl.ac.cn/Google Scholar
- Song W, Wang Y, Huang D, A rapid scene depth estimation model based on underwater light attenuation prior for underwater image restoration[C]//Advances in Multimedia Information Processing–PCM 2018: 19th Pacific-Rim Conference on Multimedia, Hefei, China, September 21-22, 2018, Proceedings, Part I 19. Springer International Publishing, 2018: 678-688.Google Scholar
- Li C, Anwar S, Hou J, Underwater image enhancement via medium transmission-guided multi-color space embedding[J]. IEEE Transactions on Image Processing, 2021, 30: 4985-5000.Google ScholarCross Ref
- Peng Y T, Cosman P C. Underwater image restoration based on image blurriness and light absorption[J]. IEEE transactions on image processing, 2017, 26(4): 1579-1594.Google ScholarDigital Library
- Wang, N., Zhou, Y., Han, F., Zhu, H., Yao, J., 2019. UWGAN: Underwater GAN for real-world underwater color restoration and dehazing. arXiv preprint arXiv: 1912.10269.Google Scholar
Index Terms
- Underwater Image Enhancement with Color Correction using Convolutional Neural Networks
Recommendations
Underwater image enhancement by color correction and color constancy via Retinex for detail preserving
AbstractIn underwater, light attenuation causes non-uniform illumination that degrades underwater image. To enhance the degraded image, we propose an underwater image enhancement method that includes color correction, color constancy, multi-...
Graphical abstractDisplay Omitted
Highlights- The formulation of color correction compensates the red and blue channels by masking.
An Underwater Color Image Enhancement Model Based on Comprehensive Processing
ICNSER2020: Proceedings of the 2nd International Conference on Industrial Control Network And System Engineering ResearchThe propagation of light in water is affected by the absorption of water and the scattering of particles in water, which leads to the problems of low contrast, blur, color distortion and so on, the traditional underwater image enhancement algorithm is ...
Retinex-inspired color correction and detail preserved fusion for underwater image enhancement
Highlights- A Retinex-inspired method for correcting color cast in underwater images.
- A ...
AbstractMany practical underwater applications need to address color cast, blurring, and low contrast of underwater images for display and further analysis. To cope with the aforementioned issues, a novel underwater image enhancement method ...
Comments