4 个月前

混合层次网络用于图像恢复

混合层次网络用于图像恢复

摘要

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

代码仓库

tombs98/mhnet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-deblurring-on-goproMHNet
PSNR: 33.04
Params (M): 17
image-deblurring-on-hide-trained-on-goproMHNet
PSNR: 30.71
single-image-deraining-on-rain100hMHNet
PSNR: 30.34
single-image-deraining-on-rain100lMHNet
PSNR: 39.47
SSIM: 0.984
single-image-deraining-on-test100MHNet
PSNR: 31.19
SSIM: 0.903
single-image-deraining-on-test1200MHNet
PSNR: 33.41
SSIM: 0.924

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
混合层次网络用于图像恢复 | 论文 | HyperAI超神经