UNDERWATER IMAGE DE-SCATTERING AND ENHANCING USING DEHAZENET AND HWD

Pan-wang Pan ； Fei Yuan ； En Cheng

《Journal of Marine Science and Technology》 26卷4期 (2018/08) Pp. 531-540

https://doi.org/10.6119/JMST.201808_26(4).0006

摘要

De-scattering and edge enhancing are critical procedures for underwater images which suffer from serious contrast attenuation, color deviation, and edge blurring. In this paper, a novel method is proposed to enhance underwater images. Firstly, a Convolutional Neural Network (CNN) is trained end-to-end to estimate the transmission map. Simultaneously, the adaptive bilateral filter is used to refine the transmission map. Secondly, a strategy based on the white balance is proposed to remove the color deviation. Laplace pyramid fusion is utilized to obtain the fusion result of the haze-free and color-corrected image. Finally, the output image is transformed into the Hybrid Wavelets and Directional Filter Banks (HWD) domain for de-noising and edge enhancing. The experimental results show that the proposed method can remove color distortion and improve the clarity of the underwater images. Objective and subjective results demonstrate that the proposed method outperforms several state-of-the- art methods in different circumstances.

關鍵字

convolutional neural network ； adaptive bilateral filter ； hybrid wavelets and directional filter banks