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

用于图像复原的混合层次网络

Mixed Hierarchy Network for Image Restoration

Beijing Normal University(北京师范大学)

Hu Gao, Depeng Dang

AI总结:

本文提出混合层次网络MHNet,通过编码器-解码器与高分辨率分支结合并简化激活和卷积设计,在去雨和去模糊任务中平衡复原质量与系统复杂度。

摘要:

图像复原是一个长期存在的底层视觉问题,例如去模糊和去雨。在图像复原过程中,不仅需要考虑复原的空间细节和上下文信息以保证质量,还需要考虑系统复杂度。尽管许多方法已经能够保证图像复原的质量,但最先进方法的系统复杂度也在不断增加。受此启发,我们提出了一种能够平衡这些相互竞争目标的混合层次网络。我们的主要贡献是一种混合层次架构,该架构在从退化图像中逐步恢复上下文信息和空间细节的同时,我们设计了块内结构以降低系统复杂度。具体来说,我们的模型首先使用编码器-解码器架构学习上下文信息,然后将其与保留空间细节的高分辨率分支相结合。为了降低该架构的系统复杂度以便于分析和比较,我们用乘法替换或移除了非线性激活函数,并采用了简单的网络结构。此外,我们将编码器-解码器中间块的空间卷积替换为全局自注意力。由此产生的紧密互连的层次架构,命名为MHNet,在多个图像复原任务上取得了显著的性能提升,包括图像去雨和去模糊。

英文摘要:

Image restoration is a long-standing low-level vision problem, e.g., deblurring and deraining. In the process of image restoration, it is necessary to consider not only the spatial details and contextual information of restoration to ensure the quality, but also the system complexity. Although many methods have been able to guarantee the quality of image restoration, the system complexity of the state-of-the-art (SOTA) methods is increasing as well. Motivated by this, we present a mixed hierarchy network that can balance these competing goals. Our main proposal is a mixed hierarchy architecture, that progressively recovers contextual information and spatial details from degraded images while we design intra-blocks to reduce system complexity. Specifically, our model first learns the contextual information using encoder-decoder architectures, and then combines them with high-resolution branches that preserve spatial detail. In order to reduce the system complexity of this architecture for convenient analysis and comparison, we replace or remove the nonlinear activation function with multiplication and use a simple network structure. In addition, we replace spatial convolution with global self-attention for the middle block of encoder-decoder. The resulting tightly interlinked hierarchy architecture, named as MHNet, delivers strong performance gains on several image restoration tasks, including image deraining, and deblurring.

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

↑