Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Teaching Tailored to Talent: Adverse Weather Restoration via Prompt Pool and Depth-Anything Constraint

About

Recent advancements in adverse weather restoration have shown potential, yet the unpredictable and varied combinations of weather degradations in the real world pose significant challenges. Previous methods typically struggle with dynamically handling intricate degradation combinations and carrying on background reconstruction precisely, leading to performance and generalization limitations. Drawing inspiration from prompt learning and the "Teaching Tailored to Talent" concept, we introduce a novel pipeline, T3-DiffWeather. Specifically, we employ a prompt pool that allows the network to autonomously combine sub-prompts to construct weather-prompts, harnessing the necessary attributes to adaptively tackle unforeseen weather input. Moreover, from a scene modeling perspective, we incorporate general prompts constrained by Depth-Anything feature to provide the scene-specific condition for the diffusion process. Furthermore, by incorporating contrastive prompt loss, we ensures distinctive representations for both types of prompts by a mutual pushing strategy. Experimental results demonstrate that our method achieves state-of-the-art performance across various synthetic and real-world datasets, markedly outperforming existing diffusion techniques in terms of computational efficiency.

Sixiang Chen, Tian Ye, Kai Zhang, Zhaohu Xing, Yunlong Lin, Lei Zhu• 2024

Related benchmarks

TaskDatasetResultRank
Raindrop RemovalRainDrop
PSNR32.66
18
Image DesnowingSnow100K L
PSNR32.37
9
Image RestorationAll-Weather Aggregate
PSNR33.41
9
Image DerainingOutdoor-Rain
PSNR31.09
9
Perception RestorationCleanBench-Real Snow Scenes
MUSIQ67.72
9
Perception RestorationCleanBench-Real Night Scenes
MUSIQ46.79
9
Perception RestorationCleanBench-Real Fog Scenes
MUSIQ0.6458
9
Perception RestorationCleanBench-Real Rain Scenes
MUSIQ62.67
9
Image DesnowingSnow100K S
PSNR37.51
8
Video RestorationMot17 Setting 3 (test)
PSNR30
8
Showing 10 of 19 rows

Other info

Follow for update