Scientific Reports (Aug 2024)

Single image deraining via wide rectangular regional blocks and dual attention complementary enhancement network

  • Yue Shen,
  • Yuduo Zhang,
  • Wentao Li,
  • Changjie Qin,
  • Yongdong Huang

DOI
https://doi.org/10.1038/s41598-024-70329-2
Journal volume & issue
Vol. 14, no. 1
pp. 1 – 14

Abstract

Read online

Abstract Rain is a common weather phenomenon, and the challenge of removing rain streaks from a single image is crucial due to its detrimental impact on image quality and the extraction of valuable background information. Existing methods commonly rely on specific assumptions regarding rain models, which restricts their ability to accommodate a wide range of real-world scenarios. To overcome this limitation, these methods often require complex optimization techniques or stepwise refinement strategies. In this paper, we propose a novel wide rectangular regional block and dual attention complementary enhancement deraining kernel prediction subnet to meet the challenge. The network called WRRDANet consists of a kernel prediction subnet and pixel-wise dilation filtering. In the kernel prediction subnet, we capture more specific contextual background information and complex pixel-wise kernels. Afterward, the learned pixel-wise multi-scale kernels from the kernel prediction subnet are used to perform dilation filtering on the original rainy image, effectively restoring richer background details by expanding the scope of deraining to a larger extent. We conducted a comprehensive evaluation using synthetic and real rainfall datasets to demonstrate the effectiveness of our approach. The results, both qualitatively and quantitatively, indicate that our approach outperforms other popular rain removal methods.