All Projects → zijundeng → R3net

zijundeng / R3net

Licence: mit
Code for the IJCAI 2018 paper "R^3Net: Recurrent Residual Refinement Network for Saliency Detection"

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to R3net

gryds
A Python package for geometric transformations of images for data augmentation in deep learning
Stars: ✭ 55 (-47.62%)
Mutual labels:  deeplearning, computervision
Salient Object Detection
This is tensorflow implementation for cvpr2017 paper "Deeply Supervised Salient Object Detection with Short Connections"
Stars: ✭ 397 (+278.1%)
Mutual labels:  deeplearning, computervision
Groundbreaking-Papers
ML Research paper summaries, annotated papers and implementation walkthroughs
Stars: ✭ 90 (-14.29%)
Mutual labels:  deeplearning, computervision
Ai Art
PyTorch (and PyTorch Lightning) implementation of Neural Style Transfer, Pix2Pix, CycleGAN, and Deep Dream!
Stars: ✭ 153 (+45.71%)
Mutual labels:  deeplearning, computervision
Bmw Yolov4 Training Automation
This repository allows you to get started with training a state-of-the-art Deep Learning model with little to no configuration needed! You provide your labeled dataset or label your dataset using our BMW-LabelTool-Lite and you can start the training right away and monitor it in many different ways like TensorBoard or a custom REST API and GUI. NoCode training with YOLOv4 and YOLOV3 has never been so easy.
Stars: ✭ 533 (+407.62%)
Mutual labels:  deeplearning, computervision
Learnopencv
Learn OpenCV : C++ and Python Examples
Stars: ✭ 15,385 (+14552.38%)
Mutual labels:  deeplearning, computervision
Solaris
CosmiQ Works Geospatial Machine Learning Analysis Toolkit
Stars: ✭ 290 (+176.19%)
Mutual labels:  deeplearning, computervision
Bmw Tensorflow Inference Api Cpu
This is a repository for an object detection inference API using the Tensorflow framework.
Stars: ✭ 158 (+50.48%)
Mutual labels:  deeplearning, computervision
Monk v1
Monk is a low code Deep Learning tool and a unified wrapper for Computer Vision.
Stars: ✭ 480 (+357.14%)
Mutual labels:  deeplearning, computervision
Vrn
👨 Code for "Large Pose 3D Face Reconstruction from a Single Image via Direct Volumetric CNN Regression"
Stars: ✭ 4,391 (+4081.9%)
Mutual labels:  deeplearning, computervision
MTMT
Code for the CVPR 2020 paper "A Multi-task Mean Teacher for Semi-supervised Shadow Detection"
Stars: ✭ 66 (-37.14%)
Mutual labels:  deeplearning, computervision
Contrastive Unpaired Translation
Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)
Stars: ✭ 822 (+682.86%)
Mutual labels:  deeplearning, computervision
Monk object detection
A one-stop repository for low-code easily-installable object detection pipelines.
Stars: ✭ 437 (+316.19%)
Mutual labels:  deeplearning, computervision
Bmw Tensorflow Training Gui
This repository allows you to get started with a gui based training a State-of-the-art Deep Learning model with little to no configuration needed! NoCode training with TensorFlow has never been so easy.
Stars: ✭ 736 (+600.95%)
Mutual labels:  deeplearning, computervision
Novel Deep Learning Model For Traffic Sign Detection Using Capsule Networks
capsule networks that achieves outstanding performance on the German traffic sign dataset
Stars: ✭ 88 (-16.19%)
Mutual labels:  deeplearning, computervision
Hellovision
Vision framework example for my article. https://medium.com/compileswift/swift-world-whats-new-in-ios-11-vision-456ba4156bad
Stars: ✭ 93 (-11.43%)
Mutual labels:  computervision
Har Keras Cnn
Human Activity Recognition (HAR) with 1D Convolutional Neural Network in Python and Keras
Stars: ✭ 97 (-7.62%)
Mutual labels:  deeplearning
Classification Of Hyperspectral Image
Classification of the Hyperspectral Image Indian Pines with Convolutional Neural Network
Stars: ✭ 93 (-11.43%)
Mutual labels:  deeplearning
Computer Science Resources
A list of resources in different fields of Computer Science (multiple languages)
Stars: ✭ 1,316 (+1153.33%)
Mutual labels:  computervision
Sert
Semantic Entity Retrieval Toolkit
Stars: ✭ 100 (-4.76%)
Mutual labels:  deeplearning

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 [paper link]

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}${N}et: Recurrent Residual Refinement Network for Saliency Detection},
     booktitle = {IJCAI},
     year = {2018}
}

Saliency Map

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.4.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. Put the trained model in ckpt/R3Net
  3. 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

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].