/LDPS

This is the code repo for "Learning Deep Relations to Promote Saliency Detection". AAAI2020

Primary LanguagePythonMIT LicenseMIT

Learning Deep Relations to Promote Saliency Detection

This is the code repo for "Learning Deep Relations to Promote Saliency Detection", accepted by AAAI2020.

Train

python3 train_mi.py --backbone mobilenetv2 --batch-size 1 --lr 0.0005 --prior picanet --iter-num 20000 --t-fg 0.9 --t-afg 0.8 --t-abg 0.3 --test-prior srm --test-dataset ecssd

Test

python3 test_mi.py --sync-bn False --distributed False --test-dataset DATASET --test-prior PRIOR --ckp ./ckp/xxxx.pth --test True --save-result True

Citing This Paper

If you find this work useful in your research, please consider citing:

@inproceedings{chen2020r,
  title={Learning Deep Relations to Promote Saliency Detection},
  author={Chen, Changrui and Sun, Xin and Hua, Yang and Dong, Junyu and Xv, Hongwei},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  year={2020}
}