Underwater image enhancement algorithm based on dark channel prior and underwater imaging model

MATEC Web of Conferences 336, 06033 (2021) CSCNS2020 https://doi.org/10.1051/matecconf/202133606033 Underwater image enhancement algorithm based on dark channel prior and underwater imaging model Zhengping Sun , Fubing Li *, and Yuying Yang School of Information and Communication Engineering, Beijing Information Science and Technology University, Beijing 100101, China Abstract. The main reason for the degradation of the underwater image is the light absorption and scattering. The images are captured in the underwater environment often have some problems such as loss of image information, low contrast, and color distortion. In order to solve the above problems, this paper proposes an image enhancement method for the underwater environment. With the help of the underwater imaging model and dark channel prior theory, a new idea of adding transmission correction and color compensation to G and B color channels is proposed. Experimental results show that, compared with the traditional methods, this method has a better effect on the underwater image with less color deviation. 1 Introduction In recent years, the application of underwater environmental monitoring and information collection is more and more extensive, so the collection and research of underwater information with clear underwater images are the most important. However, the particle molecules in the water will cause the scattering of light, and the absorption characteristics of water to light will also cause light attenuation, which will cause poor image quality, blue-green fog surface feeling, low contrast, limiting the visibility, and other defects. Underwater image quality enhancement can be divided into image restoration methods [1-3] and image enhancement methods [4-6]. The process of image restoration to enhance image quality mainly depends on a specific degradation process and a real imaging model. The process of using the image enhancement method to enhance image quality only needs to pay attention to the pixel value of the image and adjust the pixel value to optimize the overall image quality. 2 Imaging model of light in the underwater environment In the special underwater environment, the attenuation is caused by the absorption and * Corresponding author: © The Authors, published by EDP Sciences. This is an open access article distributed under the terms of the Creative Commons Attribution License 4.0 (http://creativecommons.org/licenses/by/4.0/). MATEC Web of Conferences 336, 06033 (2021) CSCNS2020 https://doi.org/10.1051/matecconf/202133606033 scattering of light by water. Absorption means that part of the light wave will be absorbed by water and converted into heat energy of water in underwater transmission, which causes energy loss. Moreover, the longer the wavelength is, the more obvious the absorption effect is. The scattering is caused by the suspended particles in the water, and the current propagation direction will be shifted. According to reference [1], the light components arriving at the imaging interface are direct component, forward scattering component, and backscattering component. The direct component only considers the attenuation of the reflected light due to the medium， the reflected light of the object in the underwater scene represents the real information of the object; the forward scattering component is the component with a small-angle deviation of the direct component; the backscattering component is different from the first two components, which is caused by the suspended particles in the underwater environment, and light passing through these particles will scatter at a large angle. Fig.1 illustrates the imaging model of the underwater environment. Considering that the forward scattering component has little effect on the phase ambiguity caused by imaging, even if forward scattering accounts for 90% of the total scattering events, it can also be ignored. Therefore, the imaging model can be simplified as the sum of direct component and backscattering component. sunlight suspended particles object imaging plane underwater scene direct component forward scattering component backscattering component Fig. 1. Underwater optical imaging model. 3 Dark channel prior algorithm Because the underwater imaging model is similar to the fog imaging model, and the contrast and clarity of the image will be reduced due to the attenuation of the medium. Therefore, the dark channel prior to the foggy environment can be applied to the underwater environment. However, the underwater complex environment is different from the fog environment, because the attenuation of light in different wavebands is variant, so the attenuation is the most serious for the red light with the longest wavelength. If the prior theory of fog image is used in the underwater environment, the value of dark channel will be very low and cannot display any information of the underwater scene. Reference [3] proposed a prior theory of dark channel for the underwater scene, namely Red Channel prior theory. To conform to the prior theory of dark channel, it is necessary to take the red channel inversely, and then split the imaging model into three color channels: 1 − I R =t R (1 − J R ) + (1 − t R )(1 − BR ,∞ ) I G = tG ⋅ J G + (1 − tG ) BG ,∞ I B = t B ⋅ J B + (1 − t B ) BB ,∞ 2 (1) MATEC Web of Conferences 336, 06033 (2021) CSCNS2020 https://doi.org/10.1051/matecconf/202133606033 The Red Channel prior is as follows: J Re d = min[min(1 − J R ), min J G , min J B ] → 0 Ω Ω Ω (2) Combined with the prior theory of Red Channel, the transmission can be expressed as: tλ = 1 − min[ min(1 − I R ) min I G min I B Ω , Ω , Ω ] 1 − BR ,∞ BG ,∞ BB ,∞ (3) Underwater background light Bλ ,∞ in the underwater environment generally takes the pixel value with the highest intensity in the image: Bλ ,∞ = max(max I λ ) I Ω (4) Combined with the above transmittance and underwater background light, the enhanced underwater image can be expressed as: = Jλ I λ − Bλ ,∞ tλ + Bλ ,∞ (5) 4 The improved image enhancement algorithm Even considering the different attenuation degrees, wavelength, and transmission of R, G, and B channels in the underwater environment, the recovered image is still unsatisfactory. To solve the problems of low contrast and loss of details, the transmission correction and color compensation are proposed in this paper: 1) In the step of Red Channel prior processing image, a correction factor is added in the process of transmission estimation. min(1 − I R ) min I G min I B 1 − min[ Ω , ε1 ⋅ Ω ,ε2 ⋅ Ω ] tλ = BG ,∞ BB ,∞ 1 − BR ,∞ (6) 2) The idea of shielding G to compensate R and B is adopted, that is adding a part of G to R and B. The purpose of color compensation can be achieved by color compensation of image J. G R − J mean J RC = J R + β1 ⋅ ( J mean ) ⋅ JG G B − J mean J BC = J B + β 2 ⋅ ( J mean )⋅ JB (7) J RC and J BC are the R and B components of the final output image. 5 Experimental res (...truncated)