arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.09912cs.CVeess.IV

利用多分辨率与多尺度注意力进行水下图像复原

Harnessing Multi-resolution and Multi-scale Attention for Underwater Image Restoration

Indian Institute of Technology Guwahati(印度古瓦哈蒂印度理工学院)

Alik Pramanick, Arijit Sur, V. Vijaya Saradhi

AI总结:

针对现有水下图像复原方法难以兼顾空间信息与语义精度的问题,提出轻量级多阶段网络Lit-Net,结合多分辨率多尺度分析、并行1×1卷积编码器与改进的通道损失函数,在EUVP数据集上的性能优于现有SOTA方法,可服务于水下自主设备与监控。

摘要:

水下图像通常会受到色彩失真、低对比度等因素的影响,给高层视觉任务带来挑战。现有的水下图像复原(UIR)方法要么以全分辨率分析输入图像,虽具备丰富的空间信息但上下文感知能力较弱;要么从高分辨率到低分辨率逐步处理,虽能得到可靠的语义信息但空间精度有所下降。为此,我们提出了一种名为Lit-Net的轻量级多阶段网络,该网络聚焦于多分辨率与多尺度图像分析以实现水下图像复原:第一阶段保留原始分辨率,第二阶段细化特征,最后阶段专注于图像重建。我们设计的新型编码器模块采用并行1×1卷积层来捕获局部信息并提升运算速度。此外,我们还引入了改进的加权彩色通道特异性l₁损失(cl₁)函数,以恢复色彩与细节信息。在公开数据集上开展的大量实验表明,我们的模型性能优于当前最新的方法,在定性和定量指标上均有显著提升:例如在EUVP数据集上,PSNR达到29.477 dB(提升1.92%),SSIM达到0.851(提升2.87%)。Lit-Net的相关贡献为水下图像增强与超分辨率提供了一种更鲁棒的方法,对水下自主航行器与监控系统具有重要意义。代码可在以下链接获取:https://github.com/Alik033/Lit-Net。

英文摘要:

Underwater imagery is often compromised by factors such as color distortion and low contrast, posing challenges for high-level vision tasks. Recent underwater image restoration (UIR) methods either analyze the input image at full resolution, resulting in spatial richness but contextual weakness, or progressively from high to low resolution, yielding reliable semantic information but reduced spatial accuracy. Here, we propose a lightweight multi-stage network called Lit-Net that focuses on multi-resolution and multi-scale image analysis for restoring underwater images while retaining original resolution during the first stage, refining features in the second, and focusing on reconstruction in the final stage. Our novel encoder block utilizes parallel $1\times1$ convolution layers to capture local information and speed up operations. Further, we incorporate a modified weighted color channel-specific $l_1$ loss ($cl_1$) function to recover color and detail information. Extensive experimentations on publicly available datasets suggest our model's superiority over recent state-of-the-art methods, with significant improvement in qualitative and quantitative measures, such as $29.477$ dB PSNR ($1.92\%$ improvement) and $0.851$ SSIM ($2.87\%$ improvement) on the EUVP dataset. The contributions of Lit-Net offer a more robust approach to underwater image enhancement and super-resolution, which is of considerable importance for underwater autonomous vehicles and surveillance. The code is available at: https://github.com/Alik033/Lit-Net.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑