FDNet:Fourier transform guided dual-channel underwater image enhancement diffusion network  

作　　者：Zhen ZHU Xiaobo LI Qianwen MA Jingsheng ZHAI Haofeng HU 

机构地区：[1]School of Marine Science and Technology,Tianjin University,Tianjin 300072,China [2]College of Mechanical Engineering and Automation,Fuzhou University,Fuzhou 350108,China

出　　处：《Science China(Technological Sciences)》2025年第1期58-74,共17页中国科学(技术科学英文版)

基　　金：supported by the National Key Research and Development Program of China(Grant No.2023YFC3108500);the National Natural Science Foundation of China(Grant Nos.62205243 and 62075161);the Key Laboratory of Intelligent Optoelectronics,Ministry of Education of China(Grant No.ZNGD20231001)。

摘　　要：Due to the scattering and absorption of light,underwater imaging yields suboptimal results characterized by low brightness,diminished image contrast,and the loss of details.Current learning-based underwater image enhancement(UIE)methods mainly focus on addressing these issues in the spatial domain,with limited attention to the Fourier frequency domain.We propose the Fourier transform guided dual-channel underwater image enhancement diffusion network(FDNet),aiming to fully leverage frequency domain information and the characteristics of a diffusion-based generation model.The amplitude-phase dualchannel not only effectively provides feedback on the energy distribution of the image but also incorporates crucial image structure and texture information.Capitalizing on these advantages,we first introduce a Fourier transform-based amplitude-phase dualchannel enhancement front-end network for UIE.This network significantly improves the quality of the network's image input,including brightness and contrast,by incorporating phase congruency edge enhancement and prior-driven amplitude mapping strategies.The preceding network mitigates the overall training challenges caused by poor input image quality.Simultaneously,the application of denoising networks based on lightweight transformers effectively improves the computational time required for each iteration during model training.The results indicate that FDNet outperforms many current learning-based UIE algorithms in restoring images across multiple real underwater image datasets,demonstrating more robust generalization capabilities in degraded underwater scenarios.The competitive performance advantage achieved in enhancing visual quality underscores the effectiveness of our method.

关 键 词：underwater image enhancement Fourier transform deep neural network phase congruency diffusion model 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]