This repo. is an official implementation of the CFIDNet , which has been accepted in the journal Neural computing and applications, 2022.
The main pipeline is shown as the following,
And some visualization results are listed
>= Pytorch 1.0.0
OpenCV-Python
[optional] matlab
pretrained resnet50 is available at (Baidu) [code:nfih]
python main.py --mode=train
python main.py --mode=test --model=./pretrained/best.pth --sal_mode="DES"
We provide the trained model file (Baidu) [code:itp2]
The saliency maps are also available (Baidu). [code:labl]
Please cite the CFIDNet in your publications if it helps your research:
@article{CHEN2022,
title = {CFIDNet: Cascaded Feature Interaction Decoder for RGB-D Salient Object Detection},
author = {Tianyou Chen and Xiaoguang Hu and Jin Xiao and Guofeng Zhang and Shaojie Wang},
journal = {Neural Computing and Applications},
year = {2022},
}