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 480×640. BER is reported for 480×640 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 480×640. For the pre-trained weights, we only provide SP+M weights, since this repository has full implementation of it. RMSE is reported for 480×640 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.

GitHub

https://github.com/naoto0804/SynShadow