/SCRN

Code of Stacked Cross Refinement Network for Edge-Aware Salient Object Detection (ICCV 2019)

Primary LanguagePython

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

image

Experiments

  1. Results on traditional datasets results1

  2. Results on SOC (attribute-based performance, structure simalarity scores), more comparison can be found in SOC Leaderboard results3

Usage

  1. Requirements
    • pytorch 0.40+
    • scipy
  2. Clone the repo
git clone https://github.com/wuzhe71/SCRN.git 
cd SCRN
  1. Train/Test
    • Train
      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
      

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)): DSSNLDFSRMAmuletDGRLBMPMPiCANet-RR3NetC2S-NetRANetCPDAFNBASNetPoolNetSCRNSIBAEGNetF3NetGCPANetMINet.

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).