DenseSR: Image Shadow Removal as Dense Prediction
Paper
β’
2507.16472
β’
Published
β’
1
Yu-Fan Lin1, Chia-ming Lee1, Chih-Chung Hsu2
1National Cheng Kung University 2National Yang Ming Chiao Tung University
If you find this project useful, please consider citing us and giving us a star.
@misc{lin2025densesrimageshadowremoval,
title={DenseSR: Image Shadow Removal as Dense Prediction},
author={Yu-Fan Lin and Chia-Ming Lee and Chih-Chung Hsu},
year={2025},
eprint={2507.16472},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2507.16472},
}
conda create -n ntire_shadow python=3.9 -y
conda activate ntire_shadow
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt
You can download WSRD dataset from here.
test_dir
βββ origin <- Put the shadow affected images in this folder
β βββ 0000.png
β βββ 0001.png
β βββ ...
βββ depth
βββ normal
output_dir
βββ 0000.png
βββ 0001.png
βββ...
git clone https://github.com/DepthAnything/Depth-Anything-V2.git
Download the pretrain model of depth anything v2
Run get_depth_normap.py to create depth and normal map.
python get_depth_normap.py
Now folder structure will be
test_dir
βββ origin
β βββ 0000.png
β βββ 0001.png
β βββ ...
βββ depth
β βββ 0000.npy
β βββ 0001.npy
β βββ ...
βββ ormal
β βββ 0000.npy
β βββ 0001.npy
β βββ ...
output_dir
βββ 0000.png
βββ 0001.png
βββ...
git clone https://github.com/facebookresearch/dinov2.git
gdown 1of3KLSVhaXlsX3jasuwdPKBwb4O4hGZD
run_test.sh to get inference results.bash run_test.sh
β 2025/08/11 Release WSRD pretrained model
β 2025/08/11 Release inference code
β 2025/07/05 Paper Accepted by ACMMM'25
β» Release training code
β» Release other pretrained model
This code repository is release under MIT License.