• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    公众号

yeezhu/SPN.pytorch: PyTorch implementation of "Soft Proposal Networks for W ...

原作者: [db:作者] 来自: 网络 收藏 邀请

开源软件名称(OpenSource Name):

yeezhu/SPN.pytorch

开源软件地址(OpenSource Url):

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

开源编程语言(OpenSource Language):

Python 43.9%

开源软件介绍(OpenSource Introduction):

PyTorch implementation of SPN

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

[Project Page] [Paper] [Supp]

[Torch code]

Requirements

Conda virtual environment is recommended: 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. Download the backbone model VGG16 (exported from caffe model) and then the model path should be SPN.pytorch/demo/models/VGG16_ImageNet.pt.

  3. Install SPN:

    cd SPN.pytorch/spnlib
    bash make.sh
  4. Run the training demo:

    cd SPN.pytorch/demo
    bash runme.sh
  5. Run the testing demo: EvaluationDemo.ipynb Figure Note: To perform bbox localization on ImageNet, firstly download the SP_GoogleNet_ImageNet model and the annotations into imagenet_eval folder. Extraxt the annotations:

    cd SPN.pytorch/demo/evaluation/imagenet_eval
    tar zxvf ILSVRC2012_bbox_val_v3.tgz    

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}
}

Acknowledgement

In this project, we reimplemented SPN on PyTorch based on wildcat.pytorch. To keep consistency with the Torch version, we use the VGG16 model exported from caffe in fcn.pytorch.




鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap