This codebase implements the system described in the paper:
Asymmetric Two-Stream Architecture for Accurate RGB-D Saliency Detection
Miao Zhang, Sun Xiao Fei, Jie Liu, Shuang Xu, Yongri Piao, Huchuan Lu. In ECCV 2020.
- Ubuntu 18
- PyTorch 1.3.1
- CUDA 10.1
- Cudnn 7.5.1
- Python 3.7
- Numpy 1.17.3
Download Link. Code: nx8x
Download Link. Code: qqsf
Firstly, you need to download the 'Testing dataset' and the pretraind checpoint we provided (Baidu Pan. Code: d2o0). Then, you need to set dataset path and checkpoint name correctly. and set the param '--phase' as "test" and '--param' as 'True' in demo.py.
python demo.py
Once the train-augment dataset are prepared,you need to set dataset path and checkpoint name correctly. and set the param '--phase' as "train" and '--param' as 'True'(loading checkpoint) or 'False'(do not load checkpoint) in demo.py.
python demo.py
If you have any questions, please contact us ([email protected]; [email protected]).