/object-detection-using-tensorflow-faster-rcnn

Tensorflow implementation of Faster R-CNN

Primary LanguagePython

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

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