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

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

Tensorflow implementation of Regions with Convolutional Neural Networks Features.
https://github.com/sirius-mhlee/object-detection-using-tensorflow-rcnn

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

Last synced: 2 months ago
JSON representation

Tensorflow implementation of Regions with Convolutional Neural Networks Features.

Awesome Lists containing this project

README

          

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

Tensorflow implementation of Regions with Convolutional Neural Networks Features.
http://www.rossgirshick.info/
Using 17 Category Flower Dataset.
http://www.robots.ox.ac.uk/~vgg/data/flowers/17/

Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation
R. Girshick, J. Donahue, T. Darrell, J. Malik
IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2014

Training Usage
-----

python TrainingModel.py train_data_file finetune_data_file max_epoch batch_size
output_cnn_model_file output_cnn_mean_file output_svm_model_file output_reg_model_file

ex)
python TrainingModel.py ./data/train_data.txt ./data/finetune_data.txt 3 3
./alexnet/alexnet_model.npy ./alexnet/alexnet_mean.txt ./svm/svm_model.npy ./bbox/bbox_model.npy

Training Result
-----

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

Detection Usage
-----

python DetectionImage.py cnn_model_file cnn_mean_file svm_model_file
reg_model_file label_file input_image_file output_image_file

ex)
python DetectionImage.py ./alexnet/alexnet_model.npy ./alexnet/alexnet_mean.txt ./svm/svm_model.npy
./bbox/bbox_model.npy ./alexnet/flower_classes.txt ./example/tulip.jpg ./example/tulip_result.jpg

Detection Result
-----