Git Product home page Git Product logo

r3net's Introduction

R3Net: Recurrent Residual Refinement Network for Saliency Detection

by Zijun Deng, Xiaowei Hu, Lei Zhu, Xuemiao Xu, Jing Qin, Guoqiang Han, and Pheng-Ann Heng

This implementation is written by Zijun Deng at the South China University of Technology.


Citation

@inproceedings{deng18r,
     author = {Deng, Zijun and Hu, Xiaowei and Zhu, Lei and Xu, Xuemiao and Qin, Jing and Han, Guoqiang and Heng, Pheng-Ann},
     title = {R$^{3}$Net: Recurrent Residual Refinement Network for Saliency Detection},
     booktitle = {IJCAI},
     year = {2018}
}

Saliency Maps

The results of salienct object detection on five datasets (ECSSD, HKU-IS, PASCAL-S, SOD, DUT-OMRON) can be found at Google Drive.

Trained model

You can download the trained model which is reported in our paper at Google Drive.

Requirement

  • Python 2.7
  • PyTorch >= 0.3.0
  • torchvision
  • numpy
  • Cython
  • pydensecrf (here to install)

Training

  1. Set the path of pretrained ResNeXt model in resnext/config.py
  2. Set the path of MSRA10K dataset in config.py
  3. Run by python train.py

The pretrained ResNeXt model is ported from the official torch version, using the convertor provided by clcarwin. You can directly download the pretrained model ported by me.

Hyper-parameters of training were gathered at the beginning of train.py and you can conveniently change them as you need.

Training a model on a single GTX 1080Ti GPU takes about 70 minutes.

Testing

  1. Set the path of five benchmark datasets in config.py
  2. Run by python infer.py

Settings of testing were gathered at the beginning of infer.py and you can conveniently change them as you need.

Useful links

r3net's People

Contributors

zijundeng avatar

Watchers

James Cloos avatar Xiaowei Hu avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.