/SeeNet

Self-Erasing Network for Integral Object Attention

Primary LanguageJupyter Notebook

SeeNet

Self-Erasing Network for Integral Object Attention in NeurIPS2018

Network architecture and more details

Please refer to our paper.

Usage

Please install Caffe first. I think you may find a great number of tutorials talking about how to install it.

cd <caffe_root>/examples
git clone https://github.com/Andrew-Qibin/SeeNet.git

ipython notebook vis.ipynb

Before you start, you also need our pretrained model (google drive).

Visual examples

Attention maps for all VOC images (10582)

You can find them here (google drive).

If you think this work is helpful, please cite

@inproceedings{hou2018selferasing,
  title={Self-Erasing Network for Integral Object Attention},
  author={Hou, Qibin and Jiang, Peng-Tao and Wei, Yunchao and Cheng, Ming-Ming},
  booktitle={NeurIPS},
  year={2018}
}