Skip to content

CAGNet: Content-Aware Guidance for Salient Object Detection

License

Notifications You must be signed in to change notification settings

Mehrdad-Noori/CAGNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CAGNet

This repository contains the Tensorflow implementation of our paper "CAGNet: Content-Aware Guidance for Salient Object Detection". The paper can be found at: [Sciencedirect] [arXiv]

Requirements

Usage

1- Clone the repository

git clone https://github.com/Mehrdad-Noori/CAGNet
cd CAGNet

2- If you want to train the model, download the following dataset and unzip it into data folder.

3- To run the training, set the arguments or use the default settings:

python train.py --backbone_model 'ResNet50' --batch_size 10 --save_dir 'save'

The backbone_model can be one of the following options: VGG16, ResNet50, NASNetMobile or NASNetLarge

4- To generate saliency maps:

python predict.py --model '/path/to/trained/model' --input_dir /path/to/input/images/directory --save_dir 'save'

You can also download and use our pre-trained models

5- Evaluation code

You can use this toolbox to compute different saliency measures such as E-measure, Weighted F-measure, MAE, PR curve ...

Pre-trained models & pre-computed saliency maps

We provide the pre-trained model and pre-computed saliency maps for DUTS-TE, ECSSD, DUT-OMRON, PASCAL-S, and HKU-IS datasets.

Quantitative Comparison

image

Qualitative Comparison

image

Any problems?

Please feel free to contact me, or raise an issue if you encounter any problems.

Citation

@article{mohammadi2020cagnet,
  title={CAGNet: Content-Aware Guidance for Salient Object Detection},
  author={Mohammadi, Sina and Noori, Mehrdad and Bahri, Ali and Majelan, Sina Ghofrani and Havaei, Mohammad},
  journal={Pattern Recognition},
  pages={107303},
  year={2020},
  publisher={Elsevier}
}

Releases

No releases published

Packages

No packages published

Languages