高级检索
范新南, 杨鑫, 史朋飞, 韩松, 辛元雪. 特征融合生成对抗网络的水下图像增强[J]. 计算机辅助设计与图形学学报, 2022, 34(2): 264-272. DOI: 10.3724/SP.J.1089.2022.18843
引用本文: 范新南, 杨鑫, 史朋飞, 韩松, 辛元雪. 特征融合生成对抗网络的水下图像增强[J]. 计算机辅助设计与图形学学报, 2022, 34(2): 264-272. DOI: 10.3724/SP.J.1089.2022.18843
Fan Xinnan, Yang Xin, Shi Pengfei, Han Song, Xin Yuanxue. Underwater Image Enhancement Based on Feature Fusion Generative Adversaral Networks[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(2): 264-272. DOI: 10.3724/SP.J.1089.2022.18843
Citation: Fan Xinnan, Yang Xin, Shi Pengfei, Han Song, Xin Yuanxue. Underwater Image Enhancement Based on Feature Fusion Generative Adversaral Networks[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(2): 264-272. DOI: 10.3724/SP.J.1089.2022.18843

特征融合生成对抗网络的水下图像增强

Underwater Image Enhancement Based on Feature Fusion Generative Adversaral Networks

  • 摘要: 针对水下图像对比度低和颜色失真等问题,提出一种特征融合生成对抗网络的水下图像增强算法.首先,对水下退化图像进行颜色校正,并以卷积神经网络提取颜色校正后图像的特征;其次,以基于U-Net的特征提取网络提取水下退化图像特征,并将其与颜色校正图像的特征融合;最后,通过卷积神经网络完成融合特征到增强图像的重构.在Underwater-ImageNet数据集上与其他算法相比,水下图像评价指标(underwater image quality measure,UIQE)提高0.308,自然图像评价指标(natural image quality evaluator,NIQE)降低0.638,增强后的水下图像对比度和清晰度提升并且颜色更真实.

     

    Abstract: Aiming at the problems of low contrast and color distortion of underwater images, an underwater image enhancement algorithm based on feature fusion generative adversarial networks is proposed. Firstly, color correction algorithm is applied to the underwater degraded image, and then the feature of color corrected image is extracted by convolution neural network. Secondly, the feature of underwater degraded image can be extracted by the feature extraction network which is based on U-Net, and then fuse it with the feature of color correction image. Finally, the convolution neural network is used to reconstruct the fusion feature to the enhanced image. The experimental results on Underwater-ImageNet dataset show that the algorithm can effectively improve the contrast and clarity of underwater degraded image, and the enhanced image color is more realistic. Compared with other algorithms on the Underwater-ImageNet dataset, the underwater image quality measure (UIQE) is increased by 0.308, the natural image quality evaluator (NIQE) is reduced by 0.638. The contrast and sharpness of the enhanced underwater image are improved and the colors are more realistic.

     

/

返回文章
返回