李耀, 于腾, 祁少华, 杨国为. 基于CGAN的自适应密集特征融合水下图像增强算法[J]. 微电子学与计算机, 2021, 38(12): 31-38. DOI: 10.19304/J.ISSN1000-7180.2021.0517
引用本文: 李耀, 于腾, 祁少华, 杨国为. 基于CGAN的自适应密集特征融合水下图像增强算法[J]. 微电子学与计算机, 2021, 38(12): 31-38. DOI: 10.19304/J.ISSN1000-7180.2021.0517
LI Yao, YU Teng, QI Shaohua, YANG Guowei. Adaptive dense feature fusion underwater image enhancement algorithm based on CGAN[J]. Microelectronics & Computer, 2021, 38(12): 31-38. DOI: 10.19304/J.ISSN1000-7180.2021.0517
Citation: LI Yao, YU Teng, QI Shaohua, YANG Guowei. Adaptive dense feature fusion underwater image enhancement algorithm based on CGAN[J]. Microelectronics & Computer, 2021, 38(12): 31-38. DOI: 10.19304/J.ISSN1000-7180.2021.0517

基于CGAN的自适应密集特征融合水下图像增强算法

Adaptive dense feature fusion underwater image enhancement algorithm based on CGAN

  • 摘要: 针对水下图像降质的问题,提出一种基于条件生成对抗网络(CGAN)的自适应密集特征融合水下图像增强算法.该算法提出一种新颖的自适应密集特征融合(ADFF)模块,通过自适应学习不同级别特征的空间重要性权重,从而促使网络从以前和现在的特征中学习更有效的特征进行融合.实验中,采用U-Net结构的生成器,将ADFF模块集成在生成器的每一级别,使用WGAN-GP对抗损失与L1L2损失的组合损失对网络模型进行约束.实验结果表明,与其他水下图像增强算法进行对比,该算法在合成和真实数据集上均取得了更优越的性能,可以生成视觉效果更好的清晰水下图像.

     

    Abstract: Aiming at the problem that underwater image degradation. This paper proposed an adaptive dense feature fusion underwater image enhancement algorithm based on conditional generative adversarial network (CGAN). The algorithm proposed a novel adaptive dense feature fusion (ADFF) module, which could prompt the network to learn more effective features from previous and current features for fusion by adaptively learning the spatial importance weights of different levels of features. In the experiment, the U-Net structure generator was used, the ADFF module was integrated at each level of the generator, and the WGAN-GP (Wasserstein GAN with gradient penalty) adversarial loss and combined loss of L1 and L2 loss was used to constrain the network model. Experimental results show that, compared with other underwater image enhancement algorithms, this algorithm achieves superior performance on both synthetic and real data sets, and can generate clear underwater images with better visual effects.

     

/

返回文章
返回