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

ShadowFormer: Global Context Helps Image Shadow Removal

About

Recent deep learning methods have achieved promising results in image shadow removal. However, most of the existing approaches focus on working locally within shadow and non-shadow regions, resulting in severe artifacts around the shadow boundaries as well as inconsistent illumination between shadow and non-shadow regions. It is still challenging for the deep shadow removal model to exploit the global contextual correlation between shadow and non-shadow regions. In this work, we first propose a Retinex-based shadow model, from which we derive a novel transformer-based network, dubbed ShandowFormer, to exploit non-shadow regions to help shadow region restoration. A multi-scale channel attention framework is employed to hierarchically capture the global information. Based on that, we propose a Shadow-Interaction Module (SIM) with Shadow-Interaction Attention (SIA) in the bottleneck stage to effectively model the context correlation between shadow and non-shadow regions. We conduct extensive experiments on three popular public datasets, including ISTD, ISTD+, and SRD, to evaluate the proposed method. Our method achieves state-of-the-art performance by using up to 150X fewer model parameters.

Lanqing Guo, Siyu Huang, Ding Liu, Hao Cheng, Bihan Wen• 2023

Related benchmarks

TaskDatasetResultRank
Shadow RemovalISTD (test)
RMSE (All)4.09
65
Shadow RemovalISTD+ (test)
RMSE (Shadow)5.2
34
Shadow RemovalSRD (test)
RMSE4.04
25
Shadow RemovalISTD
PSNR32.21
24
Shadow RemovalISTD+
PSNR35.46
22
Shadow RemovalINS synthetic (test)
PSNR28.62
18
Shadow RemovalWSRD+
PSNR25.44
13
Shadow RemovalSRD 31 (test)
PSNR (Shadow Region)36.91
12
Shadow RemovalISTD+ v1 (test)
PSNR (Shadow Region)39.67
10
Shadow RemovalSRD
RMSE4.04
8
Showing 10 of 13 rows

Other info

Code

Follow for update