Jan, 2024

统一宽度自适应动态网络用于一体化图像修复

TL;DR对比传统图像恢复方法,全能图像恢复技术引起了人们的关注,因其能恢复受不同类型和程度损害的图像。然而,现代全能图像恢复方法忽略了任务间的困难,并使用相同的网络重建受不同退化影响的图像,这导致了任务相关性的低估和计算资源分配的次优化。为了阐明任务间复杂性,我们提出了一个新概念,即将复杂图像退化表示为基本退化。在此基础上,我们提出了一种创新方法 —— 统一宽度自适应动态网络(U-WADN),由两个关键组件组成:宽度自适应主干网络(WAB)和宽度选择器(WS)。WAB 包括多个嵌套的子网络,具有不同的宽度,有助于选择最适合每个任务的计算,从而在运行时在准确性和计算效率之间取得平衡。对于不同的输入,WS 自动选择最适合的子网络宽度,考虑到任务特定和样本特定的复杂性。在各种图像恢复任务的大量实验证明,提出的 U-WADN 实现了更好的性能,同时减少了高达 32.3%的 FLOPs,并提供约 15.7%的实时加速。代码已在 https://github.com/xuyimin0926/U-WADN 公开。