naoto0804 / SynShadow

Learning from Synthetic Shadows for Shadow Detection and Removal [Inoue and Yamasaki, IEEE TCSVT 2021].
MIT License
86 stars 14 forks source link
pytorch shadow-detection shadow-removal

Learning from Synthetic Shadows for Shadow Detection and Removal (IEEE TCSVT 2021)

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

If you find this code or dataset useful for your research, please cite our paper:

@article{inoue2021learning,
  title={{Learning from Synthetic Shadows for Shadow Detection and Removal}},
  author={Inoue, Naoto and Yamasaki, Toshihiko},
  journal={IEEE Transactions on Circuits and Systems for Video Technology},
  year={2021},
  volume={31},
  number={11},
  pages={4187-4197},
  doi={10.1109/TCSVT.2020.3047977}
}

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 accidentally removed some files and cannot provide some results.