roboflow / tensorflow-object-detection-faster-rcnn

How to train an object detection model easy for free

Object Detection with MobileNetSSDv2 and Faster R-CNN and the TensorFlow Object Detection API

Thannks to https://github.com/Tony607/object_detection_demo for the intial demo!

What is contained in this repo

• A tutorial to train and use MobileNetSSDv2 with the TensorFlow Object Detection API
• A tutorial to train and use Faster R-CNN with the TensorFlow Object Detection API

What you will learn (MobileNetSSDv2)

• How to load your custom image detection from Roboflow (here we use a public blood cell dataset with tfrecord)
• Download base MobileNetSSDv2 model
• Set up training environment
• Configure training pipeline and train the model
• Export the trained model's .pb inference graph
• Use the saved model for inference

What you will learn (Faster R-CNN)

• How to load your custom image data from Roboflow (here we use a public blood cell dataset with tfrecord)
• Download base pretrained Faster R-CNN model
• Set up training environment
• Configure training pipeline and train model
• Export the trained model's .pb inference graph
• Use the saved model for inference

Resources available

• This blog post for MobileNetSSDv2 walks through the tutorial
• This blog post for Faster R-CNN walks through the tutorial
• For the MobileNetSSDv2 model tutorial
• For the Faster R-CNN model tutorial
• For reading purposes, for MobileNetSSDv2, the notebook is saved here as Tutorial_Mobilenet.ipynb
• For reading purposes, for Faster R-CNN, the notebook is also saved here as Tutorial_Faster_RCNN.ipynb

About Roboflow for Data Management

Roboflow makes managing, preprocessing, augmenting, and versioning datasets for computer vision seamless. Developers reduce 50% of their code when using Roboflow's workflow, automate annotation quality assurance, save training time, and increase model reproducibility.