Awesome
SCRN
Code repository for our paper "Stacked Cross Refinement Network for Edge-Aware Salient Object Detection", ICCV 2019 poster. Paper and supplementary material are available.
Change Log
2020.11.4:We update the predicted saliency maps of 20 algorithms on SOC test and validation sets!
Framework
Experiments
-
Results on traditional datasets
-
Results on SOC (attribute-based performance, structure simalarity scores), more comparison can be found in SOC Leaderboard
Usage
- Requirements
- pytorch 0.40+
- scipy
- Clone the repo
git clone https://github.com/wuzhe71/SCRN.git
cd SCRN
- Train/Test
- Train
- Download datasets: DUTS, DUT-OMRON, ECSSD, HKU-IS, PASCAL-S, THUR15K, SOC
- Set your dataset path, then
python train_SCRN.py
- We only use multi-scale traing for data agumentation, and the lr is set as 0.002. If you change to single-scale training, the lr should better change to 0.005.
- Test
- Download the pre-trained model from google drive or baidu yun (code: ilhx), and put it in './model/'. This model is only trained on the training set of DUTS and tested on other datasets, including SOC and test set of DUTS. Set your dataset path, then
python test_SCRN.py
- You can also download the pre-computed saliency maps from google drive or baidu yun (code: 8mty).
- Train
SOC saliency maps
In the paper, we compare SCRN with nine methods on SOC validation set. Here we provide saliency maps of 20 SOD methods on both test and validation sets (google drive or baidu yun (code: wnjp)): DSS、NLDF、SRM、Amulet、DGRL、BMPM、PiCANet-R、R3Net、C2S-Net、RANet、CPD、AFN、BASNet、PoolNet、SCRN、SIBA、EGNet、F3Net、GCPANet、MINet.
Citation
@InProceedings{Wu_2019_ICCV,
author = {Wu, Zhe and Su, Li and Huang, Qingming},
title = {Stacked Cross Refinement Network for Edge-Aware Salient Object Detection},
booktitle = {The IEEE International Conference on Computer Vision (ICCV)},
month = {October},
year = {2019}
}
Contact Us
If you have any question, please contact us (wuzh02@pcl.ac.cn).