https://github.com/mingtaoguo/faster_rcnn_tensorflow

Simple implementation of faster rcnn by TensorFlow, containing 4 step training and easy to understand
https://github.com/mingtaoguo/faster_rcnn_tensorflow

faster-rcnn object-detection rpn tensorflow

Last synced: 12 months ago
JSON representation

Simple implementation of faster rcnn by TensorFlow, containing 4 step training and easy to understand

• Host: GitHub
• URL: https://github.com/mingtaoguo/faster_rcnn_tensorflow
• Owner: MingtaoGuo
• Created: 2019-12-28T02:19:12.000Z (over 6 years ago)
• Default Branch: master
• Last Pushed: 2019-12-28T07:42:54.000Z (over 6 years ago)
• Last Synced: 2025-04-05T13:11:21.139Z (about 1 year ago)
• Topics: faster-rcnn, object-detection, rpn, tensorflow
• Language: Python
• Size: 883 KB
• Stars: 6
• Watchers: 1
• Forks: 1
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Faster_RCNN_TensorFlow

Simple implementation of faster rcnn by TensorFlow, containing 4 step training and easy to understand

# Introduction

![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/faster%20rcnn.jpg)

#### Figure draw by Mingtao Guo

# Dataset

Pascal Voc 2007

# Requirements

1. tensorflow 1.4.0

2. numpy

3. scipy

4. pillow

# How to use

1. Download the pretrained vgg16 model, and then put vgg16 model into the folder 'pretrained_VGG'.

   download address: http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz (please unzip it)

2. Download the Pascal VOC 2007 dataset.

   download address: http://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar (please unzip it)

3. Execute step1_train_rpn.py, step2_train_fast_rcnn.py, step3_finetune_rpn.py, step4_finetune_fast_rcnn.py in order.

4. Testing please execute inference.py

# Results

||||

|-|-|-|

|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/1.jpg)|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/2.jpg)|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/3.jpg)|

|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/4.jpg)|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/5.jpg)|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/6.jpg)|

|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/7.jpg)|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/8.jpg)|![](https://github.com/MingtaoGuo/Faster_RCNN_TensorFlow/blob/master/IMGS/9.jpg)|

# Reference

[1]. Ren, Shaoqing, et al. "Faster r-cnn: Towards real-time object detection with region proposal networks." Advances in neural information processing systems. 2015.

# Author

Mingtao Guo