基于改进循环对抗神经网络的船舶雾航图像去雾算法

    Ship image dehazing method based on improved cyclic adversarial generative network

    • 摘要: 雾天条件会显著降低船舶航行的可见性,影响船舶图像观测质量并威胁航行安全,因此提升雾航图像的去雾效果具有重要意义。针对传统船舶图像去雾方法在海事场景下存在的去雾不足、细节恢复能力差等问题,本文基于真实船舶航行图像构建试验数据集,提出一种融合改进循环对抗网络(CycleGAN)与注意力机制的端到端船舶图像去雾方法。该方法引入通道依赖建模机制(SE),对特征图进行压缩与聚合,增强网络的全局特征学习能力;通过跳跃连接实现多尺度通道融合,在降低计算量的同时有效捕获复杂大气条件下的雾特性,提高对不同尺度船舶目标的表征能力。此外,进一步加入通道卷积增强模块(CA)模块以强化特征选择能力,增强网络对船舶轮廓与局部细节的恢复效果。通过多项定量指标与真实雾航场景测试验证所提方法的鲁棒性,结果表明该方法在各类航行场景中均取得优于对比方法的去雾与细节恢复性能。

       

      Abstract: Foggy weather significantly degrades ship visibility and image quality, posing serious risks to navigation safety. Enhancing the dehazing performance of ship navigation images is therefore of great importance. To address the insufficient fog removal and poor detail restoration of existing dehazing methods in maritime scenarios, this study proposes an end-to-end ship image dehazing method that integrates an improved CycleGAN with attention mechanisms. A Squeeze-and-Excitation (SE) channel-attention module is introduced to aggregate feature maps, compress spatial information, and strengthen the network's ability to learn global representations. Multi-scale channel fusion is achieved through skip connections, which not only reduces computational complexity but also enables the model to better capture fog characteristics under complex atmospheric conditions and to process ship targets of different sizes. Furthermore, a Channel Attention module is incorporated to enhance feature selection and improve the restoration of ship contours and fine structural details. Quantitative evaluations and real fog-navigation experiments confirm the robustness of the proposed method, demonstrating consistent improvements over existing dehazing approaches across all tested metrics and navigation scenarios.

       

    /

    返回文章
    返回