Adaptive Depth Enhancement Network for RGB-D Salient Object Detection
This is the official implementation of "Adaptive Depth Enhancement Network for RGB-D Salient Object Detection" as well as the follow-ups. The paper has been published by IEEE Signal Processing Letters, 2025. The paper link is here.
-
Train
runpython train.py
# put pretrained models in the pretrained folder
# set '--train-root' to your training dataset folder -
Test
runpython test.py
# set '--test-root' to your test dataset folder
# set '--ckpt' as the correct checkpoint
- The pretrained models can be downloaded in Baidu Cloud (fetach code is pcmi). Then put the pretrained models such as 'resnet_50.pth' in the pretrained folder.
- The saliency maps can be approached in Baidu Cloud (fetach code is ader).
- The saliency maps can be approached in Baidu Cloud (fetach code is adet).
- Note that all testing results are provided not only including those listed in the paper.
- The evaluation tools, training and test datasets can be obtained in RGBD-SOD-tools.
@ARTICLE{yi2025adaptive,
author={Yi, Kang and Li, Yumeng and Tang, Haoran and Xu, Jing},
journal={IEEE Signal Processing Letters},
title={Adaptive Depth Enhancement Network for RGB-D Salient Object Detection},
year={2025},
volume={32},
number={},
pages={176-180},
doi={10.1109/LSP.2024.3506863}}
}