Learning from Synthetic Shadows for Shadow Detection and Removal (IEEE TCSVT 2020)
Overview
This repo is for the paper "Learning from Synthetic Shadows for Shadow Detection and Removal". We present SynShadow, a novel large-scale synthetic shadow/shadow-free/matte image triplets dataset and pipeline to synthesize it. We further show how to use SynShadow for robust and efficient shadow detection and removal.
In this repo, we provide
- SynShadow dataset:
./datasets
- SP+M implementation:
./src
- Trained models and results: below
If you find this code or dataset useful for your research, please cite our paper:
@article{inoue_2020_tcsvt,
author = {Inoue, Naoto and Yamasaki, Toshihiko},
title = {Learning from Synthetic Shadows for Shadow Detection and Removal},
journal = {IEEE Transactions on Circuits and Systems for Video Technology (TCSVT)},
volume={XX},
number={XX},
pages={XXXX-XXXX},
year={2020},
publisher={IEEE}
}
Trained Models and Results
We provide the models for shadow detection and removal for convenience. Downloaded models should be placed under ./checkpoints
.
Shadow Detection
ALl the results are in 480x640. BER is reported for 480x640 images. Below are results evaluated on ISTD test set. DSDNet++ is a modified variant of DSDNet.
Model | Train | BER | |
---|---|---|---|
DSDNet++ | SynShadow | 2.74 | results / weights |
DSDNet++ | SynShadow->ISTD | 1.09 | results / weights |
BDRAR | SynShadow | 2.74 | results / weights |
BDRAR | SynShadow->ISTD | 1.10 | results / weights |
Shadow Removal
ALl the results are in 480x640. For the pre-trained weights, we only provide SP+M weights, since this repository has full implementation of it. RMSE is reported for 480x640 images.
Model: SP+M
Train | Test | RMSE | |
---|---|---|---|
SynShadow | ISTD+ | 4.9 | results / weights / precomputed_mask |
SynShadow->ISTD+ | ISTD+ | 4.0 | results / weights / precomputed_mask |
SynShadow | SRD+ | 5.7 | results / weights / precomputed_mask |
SynShadow->SRD+ | SRD+ | 5.2 | results / weights / precomputed_mask |
SynShadow | USR | - | results / weights / precomputed_mask |
Model: DHAN
Train | Test | RMSE | |
---|---|---|---|
SynShadow->ISTD+ | ISTD+ | 4.6 | results |
SynShadow->SRD+ | SRD+ | 6.6 | results |
SynShadow | USR | - | results |
Note: we have accidentially removed some files and cannot provide some results.