Search papers, labs, and topics across Lattice.
The paper introduces Derain-Agent, a plug-and-play framework that enhances existing deraining models by dynamically adapting to complex rain degradations. It uses a Planning Network to schedule an optimal sequence of restoration tools and a Strength Modulation mechanism to apply these tools with spatially adaptive intensity. Experiments show Derain-Agent improves the performance of state-of-the-art deraining models on synthetic and real-world benchmarks by precisely correcting region-specific errors.
Forget static deraining pipelines: Derain-Agent intelligently orchestrates restoration tools with spatially adaptive intensity, breathing new life into existing models.
While deep learning has advanced single-image deraining, existing models suffer from a fundamental limitation: they employ a static inference paradigm that fails to adapt to the complex, coupled degradations (e.g., noise artifacts, blur, and color deviation) of real-world rain. Consequently, restored images often exhibit residual artifacts and inconsistent perceptual quality. In this work, we present Derain-Agent, a plug-and-play refinement framework that transitions deraining from static processing to dynamic, agent-based restoration. Derain-Agent equips a base deraining model with two core capabilities: 1) a Planning Network that intelligently schedules an optimal sequence of restoration tools for each instance, and 2) a Strength Modulation mechanism that applies these tools with spatially adaptive intensity. This design enables precise, region-specific correction of residual errors without the prohibitive cost of iterative search. Our method demonstrates strong generalization, consistently boosting the performance of state-of-the-art deraining models on both synthetic and real-world benchmarks.
