TCGNet: Type-Correlation Guidance for Salient Object Detection

Requirements

• PyTorch == 1.0.0
• TorchVision == 0.2.1
• CUDA 10.0 cudnn 7.2

Backbone Pretrained Model

Please download backbone pretrained model from https://1drv.ms/u/s!Aus8VCZ_C_33dKvqI6pBZlifgJk, and put it into backbone package.

Train

Please Run train.py

Infer

Please Run infer.py

Datasets

Please alter path in config.py and datasets.py

Saliency map

Saliency map can be downloaded from

通过百度网盘分享的文件：results 链接：https://pan.baidu.com/s/17rX2sm4N18hudgarXOHctA?pwd=369E 提取码：369E 复制这段内容打开「百度网盘APP 即可获取」