Underwater image enhancement presents unique challenges due to blurriness, color distortion, and reduced visibility caused by water properties. In this paper, we propose WRSU-Net, a W-Net architecture combined with a Residual connection and Small U-Net Block (ReSU-Block) to address these issues. The W-Net’s encoder-decoder design, enhanced by the ReSU-Block’s ability to capture multi-level features, significantly improves the clarity and structural accuracy of underwater images. By utilizing a combination of Smooth L1 Loss [3] and SSIM Loss, our approach achieves superior results in comparison to existing methods. Experimental results demonstrate that our model outperforms other state-of-the-art techniques on the LSUI dataset, achieving a PSNR of 28.624 and SSIM [13] of 0.925.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

WRSU-Net: Enhanced Underwater Image Restoration Using W-Net and ReSU-Block for Multi-scale Feature Fusion

  • Hong-Kai Chen,
  • Jun-Wei Hsieh,
  • Han-Ting Huang,
  • Meng-Yu Kao

摘要

Underwater image enhancement presents unique challenges due to blurriness, color distortion, and reduced visibility caused by water properties. In this paper, we propose WRSU-Net, a W-Net architecture combined with a Residual connection and Small U-Net Block (ReSU-Block) to address these issues. The W-Net’s encoder-decoder design, enhanced by the ReSU-Block’s ability to capture multi-level features, significantly improves the clarity and structural accuracy of underwater images. By utilizing a combination of Smooth L1 Loss [3] and SSIM Loss, our approach achieves superior results in comparison to existing methods. Experimental results demonstrate that our model outperforms other state-of-the-art techniques on the LSUI dataset, achieving a PSNR of 28.624 and SSIM [13] of 0.925.