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. 

More Great AIM Stories

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.

More Stories

OUR UPCOMING EVENTS

8th April | In-person Conference | Hotel Radisson Blue, Bangalore

Organized by Analytics India Magazine

View Event >>

30th Apr | Virtual conference

Organized by Analytics India Magazine

View Event >>

MORE FROM AIM

3 Ways to Join our Community

Discord Server

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

Telegram Channel

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

Subscribe to our newsletter

Get the latest updates from AIM