An open API service indexing awesome lists of open source software.

https://github.com/sirius-mhlee/object-detection-using-tensorflow-faster-rcnn

Tensorflow implementation of Faster R-CNN
https://github.com/sirius-mhlee/object-detection-using-tensorflow-faster-rcnn

deep-learning fast-rcnn object-detection paper-implementation python tensorflow

Last synced: about 2 months ago
JSON representation

Tensorflow implementation of Faster R-CNN

Awesome Lists containing this project

README

          

Object Detection using Tensorflow Faster R-CNN
===========================================

Tensorflow implementation of Faster R-CNN.
http://www.rossgirshick.info/
Using 17 Category Flower Dataset.
http://www.robots.ox.ac.uk/~vgg/data/flowers/17/

Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun
Neural Information Processing Systems (NIPS), 2015

Training Usage
-----

python TrainingModel.py train_data_file cnn_model_file cnn_mean_file
output_cnn_model_file output_rpn_model_file output_detection_model_file

ex)
python TrainingModel.py ./data/train_data.txt ./alexnetconv/alexnet_model.npy ./alexnetconv/alexnet_mean.txt
./alexnetconv/alexnetconv_model.npy ./rpn/rpn_model.npy ./detection/detection_model.npy

Training Result
-----

![train_result.jpg](./example/train_result.jpg)

Detection Usage
-----

python DetectionImage.py cnn_mean_file cnn_model_file rpn_model_file
detection_model_file label_file input_image_file output_image_file

ex)
python DetectionImage.py ./alexnetconv/alexnet_mean.txt ./alexnetconv/alexnetconv_model.npy ./rpn/rpn_model.npy
./detection/detection_model.npy ./alexnetconv/flower_classes.txt ./example/tulip.jpg ./example/tulip_result.jpg

Detection Result
-----