Home

Awesome

AFNet

Adaptive fusion network for RGB-D salient object detection

This repo. is an official implementation of the AFNet , which has been accepted in the journal Neurocomputing, 2023.

The main pipeline is shown as the following, AFNet

And some results are presented quantitative results qualitative results

Dependencies

>= Pytorch 1.0.0
OpenCV-Python
[optional] matlab

Datasets

Training and testing datasets are available at

(Baidu) [code:jyb2]

Training

python train.py

We provide the trained backbone Google.

Test

 python test.py

We provide the trained model file (Baidu) [code:r7wc] (Google)

The prediction results are also available (Baidu). [code:ijj9]

Citation

Please cite the AFNet in your publications if it helps your research:

@article{CHEN2022,
  title = {Adaptive fusion network for RGB-D salient object detection},
  author = {Tianyou Chen and Jin Xiao and Xiaoguang Hu and Guofeng Zhang and Shaojie Wang},
  journal = {Neurocomputing},
  year = {2023},
}

Reference

BBSNet