FSDNet
FSDNet copied to clipboard
FSDNet
Revisiting Shadow Detection: A New Benchmark Dataset for Complex World
by Xiaowei Hu, Tianyu Wang, Chi-Wing Fu, Yitong Jiang, Qiong Wang, and Pheng-Ann Heng.
Fast Shadow Detection Network
This implementation is written by Xiaowei Hu at the Chinese University of Hong Kong.
Citation
@article{hu2021revisiting,
title={Revisiting Shadow Detection: A New Benchmark Dataset for Complex World},
author={Hu, Xiaowei and Wang, Tianyu and Fu, Chi-Wing and Jiang, Yitong and Wang, Qiong and Heng, Pheng-Ann},
journal={IEEE Transactions on Image Processing},
volume={30},
pages={1925--1934},
year={2021},
publisher={IEEE}
}
Requirements
- PyTorch1.3.0
- Python3.6
- Cupy [Installation Guide]
Train and Test
-
Clone this repository:
git clone https://github.com/xw-hu/FSDNet.git
-
Train:
python3 train.py
-
Test:
python3 infer.py
CUHK-Shadow Dataset
Please find the dataset at https://github.com/xw-hu/CUHK-Shadow#cuhk-shadow-dateset.
Evaluation Function
Please find the evaluation function at https://github.com/xw-hu/CUHK-Shadow#cuhk-shadow-evaluation.