Skip to content

Global Context-Aware Progressive Aggregation Network for Salient Object Detection

Notifications You must be signed in to change notification settings

JosephChenHub/GCPANet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Global Context-Aware Progressive Aggregation Network for Salient Object Detection

This repo. is an implementation of GCPANet , which is accepted for presentation in AAAI 2020.

GCPANet

vis

dependencies

>= Ubuntu 16.04 
>= Python3.5
>= Pytorch 1.0.0
OpenCV-Python

preparation

  • download the official pretrained model (Google drive) of ResNet-50 implemented in Pytorch if you want to train the network again.
  • download or put the RGB saliency benchmark datasets (Google drive) in the folder of data for training or test.

training

you may revise the TAG and SAVEPATH defined in the train.py. After the preparation, run this command

python3 train.py

make sure that the GPU memory is enough (the original training is conducted on a NVIDIA RTX (24G) card with the batch size of 32).

test

After the preparation, run this commond

 python3 test.py model/model-xxxxx.pt

We provide the trained model file (Google drive), and run this command to check its completeness:

cksum model-100045448.pt 

you will obtain the result 100045448 268562671 model_100045448.pt. The saliency maps are also available (Google drive).

evaluation

We provide the evaluation code in the folder "eval_code" for fair comparisons. You may need to revise the algorithms , data_root, and maps_root defined in the main.m. The saliency maps of the competitors are provided (Google drive).

About

Global Context-Aware Progressive Aggregation Network for Salient Object Detection

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published