PyTorch implementation of SPN

Soft Proposal Networks for Weakly Supervised Object Localization, ICCV 2017.

[Project Page] [Paper] [Supp]

[Torch code]
[Caffe code] comming soon...

Requirements

Conda virtual environment is recommended: bash conda env create -f environment.yml

• Python3.5

• PyTorch: conda install pytorch torchvision -c soumith

• Packages: torch, torchnet, numpy, tqdm

Usage

1. Clone the SPN repository:

   git clone https://github.com/yeezhu/SPN.pytorch.git

2. Install SPN:

   cd SPN.pytorch/spnlib
   bash make.sh

3. Run the demo (implemented based on wildcat.pytorch):

   cd SPN.pytorch/demo
   bash runme.sh

Citation

If you use the code in your research, please cite:

@INPROCEEDINGS{Zhu2017SPN,
    author = {Zhu, Yi and Zhou, Yanzhao and Ye, Qixiang and Qiu, Qiang and Jiao, Jianbin},
    title = {Soft Proposal Networks for Weakly Supervised Object Localization},
    booktitle = {ICCV},
    year = {2017}
}