Underwater fish image enhancement method based on color correction
DOI:
https://doi.org/10.54097/hset.v1i.498Keywords:
Image enhancement, color correction, CIE-Lab, target recognition, YOLOv5Abstract
Due to the absorption and scattering of light propagation underwater, the captured underwater images often have problems such as color bias, low contrast and poor clarity, resulting in low accuracy of underwater fish identification. To address this problem, this paper proposes an underwater fish image enhancement method based on color correction to enhance the acquired fish images and improve the accuracy of fish target recognition. Firstly, color correction is achieved by stretching the L component and changing the a and b components of the CIE-Lab color space to improve image sharpness. Then the colors of the R-G-B channels of the image are equalized to reduce the color bias. Finally, the histograms of the three R-G-B channels are redistributed. Comparison experiments were conducted with existing methods on a self-built fish image dataset, and the enhanced images were analyzed from both subjective and objective evaluations. The results showed that the enhancement effect of the method in this paper is better than other methods. Finally, the comparison experiments of target recognition before and after image enhancement were conducted on YOLOv5, and the results showed that the enhanced image target recognition accuracy was 99.8%, which was 1.2 percentage points higher than that before enhancement; the average accuracy mAP was 94.5%, which was 5.6 percentage points higher than that before image enhancement. The method in this paper can effectively improve the problems of underwater images and provide technical support for underwater target recognition.
Downloads
References
M. J. Islam, Y. Xia and J. Sattar, "Fast Underwater Image Enhancement for Improved Visual Perception," in IEEE Robotics and Automation Letters, vol. 5, no. 2, pp. 3227-3234, April 2020.
Y. Li, H. Lu, J. Li, X. Li, Y. Li and S. Serikawa, "Underwater image de-scattering and classification by deep neural network," in Computers and Electrical Engineering, vol. 54, PP. 54:68-77. August 2016.
X. Zhao, T. Jin and S. Qu, "Deriving inherent optical properties from background color and underwater image enhancement," in Ocean Engineering, vol. 94, pp. 163-172. June 2015.
Y. Zhang, F. Yang and W. He, "An approach for underwater image enhancement based on color correction and dehazing," in International Journal of Advanced Robotic Systems, vol. 17, no. 5, Sep. 2020.
K. Iqbal, R. A. Salam, A. Osman and A. Z. Talib, "Underwater Image Enhancement Using an Integrated Colour Model," in IAENG International Journal of Computer Science, vol. 34 no. 2, pp. 239-244, June 2007.
A. S. A. Ghani, N. A. M. Isa, "Underwater image quality enhancement through composition of dual-intensity images and Rayleigh-stretching," 2014 IEEE Fourth International Conference on Consumer Electronics Berlin (ICCE-Berlin), 2014, pp. 219-220.
D. Huang, Y. Wang, W. Song, J. Sequeira and S. Mavromatis, "Shallow-water Image Enhancement Using Relative Global Histogram Stretching Based on Adaptive Parameter Acquisition," in International Conference on Multimedia Modeling, PP. 453-465, 2018.
C. Wang, X. Xue, H. Li, G. Y. Zhang, H. R. Wang and L. Zhao, "Underwater image enhancement based on color correction and improved two-dimensional gamma function," in Journal of Electronic Measurement and Instrumentation, vol. 35, no. 2, pp. 171-178, Feb. 2021.
W. Zhang, X. Pan, X. Xie, L. Li and Z. Wang, "Color correction and adaptive contrast enhancement for underwater image enhancement," in Computers & Electrical Engineering, vol. 91, no, 1, pp. 106981, March 2021.
Y. Peng, P. C. Cosman, "Underwater Image Restoration Based on Image Blurriness and Light Absorption," in IEEE Transactions on Image Processing, vol. 26, no. 4, pp. 1579-1594, April 2017.
W. Song, Y. Wang, D. Huang and D. Tjondronegoro, "A Rapid Scene Depth Estimation Model Based on Underwater Light Attenuation Prior for Underwater Image Restoration," in Lecture Notes in Computer Science, pp. 678-688, Sep. 2018.
Y. Wang, S. Yin and Z. Lv, "Adaptive background light estimation with non-local a priori underwater image recovery," in Optical Precision Engineering, vol. 27, no. 2, pp. 499-510, Feb. 2019.
C. Dai, M. Lin, X. Wu, Z. Wang and Z. Guan, "Single underwater image restoration by decomposing curves of attenuating color," in Optics and Laser Technology, vol. 123, no. C, pp. 105947-105947, March 2019.
W. Song, Y. Wang, D. Huang, Q. He and Z. Wang, "Underwater image enhancement combining background light fusion and underwater dark channel a priori and color balance," in Pattern Recognition and Artificial Intelligence, vol. 31, no. 9, pp. 856-868, Sep. 2018.
C. Yang, B. Liu, "Image front-back view separation based on Lab color space texture features," in Advances in Laser and Optoelectronics, vol. 56, no. 12, pp. 59-64, Jan. 2019.
J. Guo, C. Li, C. Guo and S. Chen, "Research progress on underwater image enhancement and restoration methods," in Chinese Journal of Graphics, vol. 22, no. 3, pp. 273-287, March 2017.
K. Iqbal, M. Odetayo, A. James, Rosalina Abdul Salam and Abdullah Zawawi Hj Talib, "Enhancing the low quality images using Unsupervised Colour Correction Method," 2010 IEEE International Conference on Systems, Man and Cybernetics, 2010, pp. 1703-1709.
A. X. Dai, "Research and implementation of automatic white balance algorithm for image processing," in Soochow University,2019.
C. Li, J. Guo, R. Cong, Y. Pang and B. Wang, "Underwater Image Enhancement by Dehazing With Minimum Information Loss and Histogram Distribution Prior," in IEEE Transactions on Image Processing, vol. 25, no. 12, pp. 5664-5677, Dec. 2016.
T. Li, J. Yang, H. Zhou and T. Lian, "Research on the method of underwater image enhancement," in Journal of Physics: Conference Series, vol. 1982, no. 1, July 2021.
P. Karen, G. Chen and A. Sos, "Human-Visual-System-Inspired Underwater Image Quality Measures," in IEEE Journal of Oceanic Engineering, vol. 41, no. 3, pp. 541-551, Jan. 2016.
B. Yan, P. Fan, X. Lei, Z. Liu and F. Yang, "A Real-Time Apple Targets Detection Method for Picking Robot Based on Improved YOLOv5, " in Remote Sensing, vol. 13, no. 9, pp. 1619, April 2021.
Downloads
Published
Issue
Section
License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
