Skip to content
Code for the ECCV 2018 paper "Bidirectional Feature Pyramid Network with Recurrent Attention Residual Modules for Shadow Detection"
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
ckpt/BDRAR add the code Aug 3, 2018
resnext add the code Aug 3, 2018
README.md update README.md Aug 3, 2018
config.py add the code Aug 3, 2018
dataset.py add the code Aug 3, 2018
infer.py add the code Aug 3, 2018
joint_transforms.py add the code Aug 3, 2018
misc.py add the code Aug 3, 2018
model.py fix module name inconsistency Aug 5, 2018
train.py add the code Aug 3, 2018

README.md

Bidirectional Feature Pyramid Network with Recurrent Attention Residual Modules for Shadow Detection

by Lei Zhu^, Zijun Deng^, Xiaowei Hu, Chi-Wing Fu, Xuemiao Xu, Jing Qin, and Pheng-Ann Heng (^ joint 1st authors)

This implementation is written by Zijun Deng at the South China University of Technology.


Citation

@inproceedings{zhu18b,
     author = {Zhu, Lei and Deng, Zijun and Hu, Xiaowei and Fu, Chi-Wing and Xu, Xuemiao and Qin, Jing and Heng, Pheng-Ann},
     title = {Bidirectional Feature Pyramid Network with Recurrent Attention Residual Modules for Shadow Detection},
     booktitle = {ECCV},
     year = {2018}
}

Shadow Maps

The results of shadow detection on SBU and UCF can be found at Google Drive.

Trained Model

You can download the trained model which is reported in our paper at Google Drive.

Requirement

  • Python 2.7
  • PyTorch 0.4.0
  • torchvision
  • numpy
  • Cython
  • pydensecrf (here to install)

Preparation

  1. Set the path of pretrained ResNeXt model in resnext/config.py
  2. Set the path of SBU dataset in config.py

The pretrained ResNeXt model is ported from the official torch version, using the convertor provided by clcarwin. You can directly download the pretrained model ported by me.

Usage

Training

  1. Run by python train.py

Hyper-parameters of training were gathered at the beginning of train.py and you can conveniently change it as you need.

Training a model on a single GTX 1080Ti GPU takes about 40 minutes.

Testing

  1. Put the trained model in ckpt/BDRAR
  2. Run by python infer.py
You can’t perform that action at this time.