Adaptive multi-modal prompting for universal image restoration amidst diverse degradations
摘要
The field of image restoration focuses on recovering high-quality visual content from degraded inputs. In practical applications, images often encounter complex and varied forms of degradation, posing significant challenges for existing restoration models. To address this, we propose AM-PromptIR, an adaptive multi-modal image restoration framework that leverages implicit correlations between degradation types to guide the restoration process. Specifically, we propose an Adaptive Multi-Modal Prompt Generation that leverages a language-image model to capture the implicit relationships between various degradation types. Additionally, we introduce Simplified Mixed Attention Mechanism and Feature Enhancement and Fusion Module to improve feature extraction and fusion, respectively. By 9.2% in dehazing tasks did AM-PromptIR improve PSNR compared to the PromptIR baseline model; by 5.3% in deraining operations was the PSNR enhancement; and across various degraded tasks, an average increase of 3.0% in PSNR was achieved. These findings highlight the robustness and effectiveness of our approach in handling diverse image degradations. Our code is available here: https://github.com/buuzhangsen/AM-PromptIR.