Task Adaptive Network for Image Restoration With Combined Degradation Factors
Existing methods have achieved excellent performance on image restoration, but most of them are designed for one type of degradation. However, the weather is complex in the real world. So networks designed for single tasks are usually difficult to apply. Therefore, we propose a task-adaptive attention module to enable the network to restore images with multiple degradation factors. The task-adaptive attention module mainly includes three parts: Task-Adaptive sub-network, Task Channel Attention, and Task Operation Attention. To evaluate the model, we construct a mixed degradation factors dataset that combines three degradation factors of rain, haze, and raindrop. The experimental results show that our method not only better restores images with mixed degradation factors, but also show competitive results compared to the state-of-the-art models of each task.
Code (1)
Tasks
Image RestorationRain RemovalSingle Image DerainingSingle Image Haze RemovalSimilar Papers 제목 키워드 기반
UniRestorer: Universal Image Restoration via Adaptively Estimating Image Degradation at Proper Granularity
Recently, considerable progress has been made in allin-one image restoration. Generally, existing methods can be degradation-agnostic or degradation-aware. However, the former are limited in leveraging degradation-specif…
Image RestorationMixture-of-ExpertsAdaptive Blind All-in-One Image Restoration
Blind all-in-one image restoration models aim to recover a high-quality image from an input degraded with unknown distortions. However, these models require all the possible degradation types to be defined during the tra…
5-Degradation Blind All-in-One Image RestorationAllBlind All-in-One Image RestorationImage RestorationLearning Frequency-Aware Dynamic Transformers for All-In-One Image Restoration
This work aims to tackle the all-in-one image restoration task, which seeks to handle multiple types of degradation with a single model. The primary challenge is to extract degradation representations from the input degr…
AllDeblurringDenoisingImage Restoration+1UIR-LoRA: Achieving Universal Image Restoration through Multiple Low-Rank Adaptation
Existing unified methods typically treat multi-degradation image restoration as a multi-task learning problem. Despite performing effectively compared to single degradation restoration methods, they overlook the utilizat…
Image RestorationMulti-Task LearningTransfer LearningUnified Image RestorationDynamic Degradation Decomposition Network for All-in-One Image Restoration
Currently, restoring clean images from a variety of degradation types using a single model is still a challenging task. Existing all-in-one image restoration approaches struggle with addressing complex and ambiguously de…
AllImage Restoration