Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Model

arXiv:2403.11157v1 Announce Type: new
Abstract: Universal image restoration is a practical and potential computer vision task for real-world applications. The main challenge of this task is handling the different degradation distributions at once. Existing methods mainly utilize task-specific conditions (e.g., prompt) to guide the model to learn different distributions separately, named multi-partite mapping. However, it is not suitable for universal model learning as it ignores the shared information between different tasks. In this work, we propose an advanced selective hourglass …

arxiv cs.cv diffusion diffusion model image image restoration mapping type universal