TensorFlow Object Detection API Officially Supports TF2: Google

Earlier this month Google announced that the TF Object Detection API (OD API) officially supports TensorFlow 2. This comes as the tech giant has been working on making the TF ecosystem more compatible with frequently used models and libraries. 

The company has been migrating TF Object Detection API models to be TensorFlow 2 compatible since a year, which was evident in the Object Detection API GitHub repository, since the last few months.

Google announced that it would include the following:

  • Eager-mode compatible binaries
  • A Keras-based suite of TF2 compatible models and new architectures such as CenterNet and EfficientDet 
  • COCO pre-trained weights for all of the models provided as TF2 style object-based checkpoints
  • Access to DistributionStrategies for distributed training
  • Colab demonstrations of eager mode compatible few-shot training and inference
  • Support for keypoint estimation, data augmentation support, better visualizations, COCO evaluation

The company highlighted that this move aims at exposing users to the benefits of TF2 and Keras while being able to use TF1. “We believe that there might be many teams out there grappling with similar migration projects, so we thought that a few words about our thought process and approach here might be useful even for non-object-detection TensorFlow users,” it said.

With this, there are three kinds of users in the codebase — new users, existing TF1 users who want to migrate to TF2 and existing TF1 users who do not want to migrate just yet. 

Google also shared a detailed strategy about how it aims to benefit all three categories of users. It said it would refactor low-level core and meta-architecture to work in both TF1 and TF2, treat feature extractors/backbones as being specific to either TF1 or TF2, leverage community-maintained existing backbone implementations, separate front-end binaries for TF1 and TF2, and more.

The company plans to continue to migrate large-scale codebases from TF1 to TF2 over the next few months. It also plans to offer a more integrated, end-to-end experience for researchers in the TF2 ecosystem.

TF2 which was released in September 2019 has brought about many changes such as adopting Keras as the official high-level API for defining models. 

Download our Mobile App

Srishti Deoras
Srishti currently works as Associate Editor at Analytics India Magazine. When not covering the analytics news, editing and writing articles, she could be found reading or capturing thoughts into pictures.

Subscribe to our newsletter

Join our editors every weekday evening as they steer you through the most significant news of the day.
Your newsletter subscriptions are subject to AIM Privacy Policy and Terms and Conditions.

Our Recent Stories

Our Upcoming Events

3 Ways to Join our Community

Telegram group

Discover special offers, top stories, upcoming events, and more.

Discord Server

Stay Connected with a larger ecosystem of data science and ML Professionals

Subscribe to our Daily newsletter

Get our daily awesome stories & videos in your inbox