The TensorFlow Object Detection API has been upgraded to TensorFlow 2.0. We discuss here what the new library means for computer vision developers and why we are so excited about the new TensorFlow 2.0 Object Detection library!

TensorFlow2 meets the Object Detection API

Furthermore, we are excited to announce Roboflow's model library integration with the TensorFlow 2.0 Object Detection API so you can quickly train TensorFlow 2 computer vision models on your own custom data.

Train TensorFlow 2 Object Detection Models

Announcing Roboflow's integration with the TensorFlow 2 Object Detection Library for Custom Object Detection. You'll have a trained TensorFlow 2 model on your custom data in minutes by changing one line of code for data import.

What is the TensorFlow Object Detection API?

Computer vision developers have been using the TensorFlow Object Detection API to train and deploy object detection models since its release in 2017. The TensorFlow Object Detection API contains a model zoo of the original deep learning object detection models.

Object detection models in the TF1 OD model zoo

The Tensor Flow Object Detection API  model zoo provides a wide suite of object detection models that you can deploy to your custom dataset and build from, similar to the Detectron2 computer vision library from Facebook.

With the Tensor Flow Object Detection API allows you to quickly try out new architectures in the TensorFlow ecosystem and offers developers deployment solutions by providing model export scripts to .pb protobuf files that contain the inference graph definition. You can then export these models to venues such as TF Lite or TFJS.

Looking for TensorFlow Object Detection 1.5? At Roboflow, we published a tutorial on training TensorFlow Object Detection API 1.5 models.

What's New in the TensorFlow 2 Object Detection API?

The hype around the TensorFlow 2.0 Object Detection API is real.

Inclusion of EfficientDet Models

The feature we are most excited about at Roboflow in the TensorFlow 2 Object Detection library is the inclusion of EfficientDet family of models in the TensorFlow 2 OD Model Zoo.

EfficientDet state of the art object detection performance - now in TF2 OD API!

EfficientDet is an object detection model written by Google Brain, designed to trade of model size and inference time for performance on the COCO standard dataset. EfficientDet is a family of models scaling in size, all released in the TF 2 OD model zoo.

Models in the TF2 OD model zoo, notably EfficientDet

We implemented a tutorial on how to train EfficientDet in PyTorch. But this offering only scaled to the smallest EfficientDet model. With the new TensorFlow 2 Object Detection tutorial on how to train EfficientDet, you can scale to larger EfficientDet models.

Eager Execution of TensorFlow Operations

Another large feature in the new TensorFlow 2.0 Object Detection API is the eager execution of TensorFlow graph functions. In the original TensorFlow you would build a graph, create a session, and then execute operations. The graph abstraction ended up being quite a mess to develop in and debug, so TensorFlow 2 worked to redefine the programming environment as imperative, so you can define and execute variables as the script runs along.

Training Distribution Strategies

The library boasts the ability to train across multiple GPU or TPU resources allowing you to massively scale up your training jobs on large datasets spread across multiple TFRecord shards.

Get started with the TF2 OD Library Today!

Now that this new technology is available, let's get it out there!

We have put together a tutorial on how you can train TensorFlow2 Object Detection models on your own dataset in minutes, just by changing one line of code for your dataset's TFRecord import.

See this tutorial on how to train TensorFlow2 Object Detection Models 🚀