DPNet

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

★ 9 Open GitHub ↗

Code for the Neurocomputing 2020 paper "Dual pyramid network for salient object detection"

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Dual pyramid network for salient object detection

by Xuemiao Xu^, Jiaxing Chen^, Huaidong Zhang\*, and Guoqiang Han\* (^ joint 1st author, * joint corresponding author)[[paper link](https://www.sciencedirect.com/science/article/pii/S0925231219313451?via%3Dihub)]

This implementation is written by Jiaxing Chen at the South China University of Technology.

## Citation

@article{xu2020dual,

&nbsp;&nbsp;&nbsp;&nbsp;  title={Dual pyramid network for salient object detection},

&nbsp;&nbsp;&nbsp;&nbsp;  author={Xu, Xuemiao and Chen, Jiaxing and Zhang, Huaidong and Han, Guoqiang},

&nbsp;&nbsp;&nbsp;&nbsp;  journal={Neurocomputing},

&nbsp;&nbsp;&nbsp;&nbsp;  volume={375},

&nbsp;&nbsp;&nbsp;&nbsp;  pages={113--123},

&nbsp;&nbsp;&nbsp;&nbsp;  year={2020},

&nbsp;&nbsp;&nbsp;&nbsp;  publisher={Elsevier}    
}

## Saliency Map

The results of DPNet on six RGB saliency datasets (ECSSD, HKU-IS, PASCAL-S, SOD, DUT-OMRON, DUTS-TE) and three RGB-D saliency  datasets (NLPR, NJUD, STEREO) can be found at [Google Drive](https://drive.google.com/open?id=1IC4igBWLQvLk8cvsJF50w272vNF9i8H2).

## Trained Model

You can download the trained model which is reported in our paper at  [Google Drive](https://drive.google.com/file/d/1w79MKDV59lKtLohcK91C-9w9FL9q_-Vc/view?usp=sharing).

## Requirement

- Python 2.7
- PyTorch 0.4.0
- torchvision
- numpy
- Cython
- pydensecrf ([here](https://github.com/Andrew-Qibin/dss_crf) to install)

## Training

1. Set the path of pretrained resnet model in resnet/config.py
2. Set the path of DUTS-TR dataset in config.py
3. Run by `python train.py`

*Hyper-parameters* of training were gathered at the beginning of *train.py* and you can conveniently change them as you need.

## Testing

1. Set the path of six benchmark datasets in config.py
2. Put the trained model in ckpt/dpnet
3. Run by `python infer.py`

*Settings* of testing were gathered at the beginning of *infer.py* and you can conveniently change them as you need.

## Dataset links

- [ECSSD](http://www.cse.cuhk.edu.hk/leojia/projects/hsaliency/dataset.html), [HKU-IS](https://sites.google.com/site/ligb86/hkuis), [PASCAL-S](http://cbi.gatech.edu/salobj/), [SOD](http://elderlab.yorku.ca/SOD/), [DUT-OMRON](http://ice.dlut.edu.cn/lu/DUT-OMRON/Homepage.htm), [DUTS](http://saliencydetection.net/duts/) : the six benchmark datasets

DPNet

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README