单模型多任务图像复原
一种基于网格结构和残差密集块的新型Transformer骨干网络,通过增强注意力和局部信息强化,在多种恶劣天气图像复原任务中达到最优性能。
Tao, Wang · Zhang, Kaihao · Ziqian, Shao · Luo, Wenhan · Stenger, Bj̈orn · 路通 · Tae-Kyun, Kim · Wei, Liu · Li, Hongdong
International Journal of Computer Vision 2024
通过采样器和紧凑自注意力阶段,减少了自注意力的计算量,适合高分辨率图像复原。
残差密集Transformer块(RDTB)融合了前序和当前局部特征,网络能学到更有效的表示,输出更清晰。
在图像去雨、去雾、去雨去雾、去雪和多天气复原五个任务上达到最优,单模型覆盖多种场景。