/DCF

Code for CVPR 2021 paper. "Calibrated RGB-D Salient Object detection".

Primary LanguagePythonMIT LicenseMIT

DCF

Code repository for our paper entilted "Calibrated RGB-D Salient Object Detection" accepted at CVPR 2021.

📔 Source Code

> Requirment

  • pytorch 1.0.0+; torchvision; PIL; numpy

> Usage

1. Inference Phase

Our saliency maps.

1Saliency Maps, (fetch code is j93d), by our DCF trained on NJUD & NLPR (2185).

2Saliency Maps, (fetch code is aeq0), by our DCF trained on NJUD & NLPR & DUT (2985).

  • Notice that, the depth map is unified, which means that the closer region is closer to 1, and the farther region is closer to 0. The testset results of the new dataset ReDWeb-S can be downloaded in here (fetch code is likm).

Our pre-trained model for inferring your own dataset.

1】Download the pre-trained model, (fetch code is 2t7g), which is trained on NJUD & NLPR & DUT. Or the another model, (fetch code is epp9), which is trained on NJUD & NLPR.

2】Set the data path and ckpt_name in demo_test.py, correctly.

3】Run python demo_test.py to obtain the saliency maps.

2. Training Phase

1】Stage 1: Run python demo_train_pre.py, which performs the Depth Calibration Strategy.

2】Stage 2: Run python demo_train.py, which performs the Fusion Strategy.

> Evaluation/Training Setup

Acknowledgement

We thank all reviewers for their valuable suggestions. At the same time, thanks to the large number of researchers contributing to the development of open source in this field, particularly, Deng-ping Fan, Runmin Cong, Tao Zhou, etc.

Our feature extraction network is based on CPD backbone.

Bibtex

@InProceedings{Ji_2021_DCF,
    author    = {Ji, Wei and Li, Jingjing and Yu, Shuang and Zhang, Miao and Piao, Yongri and Yao, Shunyu and Bi, Qi and Ma, Kai and Zheng, Yefeng and Lu, Huchuan and Cheng, Li},
    title     = {Calibrated RGB-D Salient Object Detection},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2021},
    pages     = {9471-9481}
}

Contact Us

If you have any questions, please contact us ( wji3@ualberta.ca ).