/ITSD-pytorch

code for CVPR 2020 paper "Interactive Two-Stream Decoder for Accurate and Fast Saliency Detection"

Primary LanguagePython

ITSD-pytorch

Code for CVPR 2020 paper "Interactive Two-Stream Decoder for Accurate and Fast Saliency Detection"

Saliency maps can be download at: VGG (Baidu Yun [gf1i]), Resnet (Baidu Yun [sanf])

Usage:

Training:

python3 train.py --sub=[job_name] --ids=[gpus] 

Testing:

python3 test.py --sub=[job_name] --ids=[gpus] 

Notice:

Our evaluation code is slightly lower than public evaluation. Please use this code to get the results in our paper.