Hierarchical prior-guided quality assessment method for underwater images

Abstract

Many underwater image enhancement (UIE) algorithms have been proposed to improve the quality of the underwater image. However, the lack of effective objective image quality assessment (IQA) methods has hindered the further development of UIE algorithms. Unlike traditional distorted images, underwater images usually contain natural distortions such as color cast and fog effects induced by light absorption and scattering. In addition, due to the complexity of underwater scenes, the enhanced results generated using existing UIE algorithms may have artificial distortions such as under-enhancement, over-enhancement, and detail loss. Thus, it is difficult for traditional IQA methods to comprehensively capture quality-related features in complicated underwater conditions. Based on this, this paper proposes a new hierarchical prior-guided quality assessment method for underwater images with the aid of deep learning technology, which first decomposes the underwater image into the base layer and detail layer. Then, the two layers are taken as input of feature extraction network for extracting the luminance information and texture information of the image, respectively. Meanwhile, two pseudo-reference images are used to guide model training so as to obtain luminance and texture features more accurately and efficiently. Finally, the two types of features are fused to obtain the quality score of the underwater image. Extensive experiments have been conducted on two publicly available underwater image datasets, and the experimental results show that the proposed method outperforms other underwater image quality assessment methods.