Object Detection and Tracking

With ML Kit's on-device Object Detection and Tracking API, you can detect and track objects in an image or live camera feed.

Optionally, you can classify detected objects, either by using the coarse classifier built into the API, or using your own custom image classification model. See Using a custom TensorFlow Lite model for more information.

Because object detection and tracking happens on the device, it works well as the front end of the visual search pipeline. After you detect and filter objects, you can pass them to a cloud backend, such as Cloud Vision Product Search.

iOS Android

Key capabilities

  • Fast object detection and tracking Detect objects and get their locations in the image. Track objects across successive image frames.
  • Optimized on-device model The object detection and tracking model is optimized for mobile devices and intended for use in real-time applications, even on lower-end devices.
  • Prominent object detection Automatically determine the most prominent object in an image.
  • Coarse classification Classify objects into broad categories, which you can use to filter out objects you're not interested in. The following categories are supported: home goods, fashion goods, food, plants, and places.
  • Classification with a custom model Use your own custom image classification model to identify or filter specific object categories. Make your custom model perform better by leaving out background of the image.

Example results

Tracking the most prominent object across images

The example below shows the tracking data from three successive frames with the default coarse classifier provided by ML Kit.

Tracking ID 0
Bounds (95, 45), (496, 45), (496, 240), (95, 240)
Category PLACE
Classification confidence 0.9296875
Tracking ID 0
Bounds (84, 46), (478, 46), (478, 247), (84, 247)
Category PLACE
Classification confidence 0.8710938
Tracking ID 0
Bounds (53, 45), (519, 45), (519, 240), (53, 240)
Category PLACE
Classification confidence 0.8828125

Photo: Christian Ferrer [CC BY-SA 4.0]

Multiple objects in a static image

The example below shows the data for the four objects detected in the image with the default coarse classifier provided by ML Kit.

Object 0
Bounds (1, 97), (332, 97), (332, 332), (1, 332)
Category FASHION_GOOD
Classification confidence 0.95703125
Object 1
Bounds (186, 80), (337, 80), (337, 226), (186, 226)
Category FASHION_GOOD
Classification confidence 0.84375
Object 2
Bounds (296, 80), (472, 80), (472, 388), (296, 388)
Category FASHION_GOOD
Classification confidence 0.94921875
Object 3
Bounds (439, 83), (615, 83), (615, 306), (439, 306)
Category FASHION_GOOD
Classification confidence 0.9375

Using a custom TensorFlow Lite model

The default coarse classifier is built for five categories, providing limited information about the detected objects. You might need a more specialized classifier model that covers a narrower domain of concepts in more detail; for example, a model to distinguish between species of flowers or types of food.

As such, this API supports the use of custom image classification models, from a wide range of sources. You can use pre-trained models from TensorFlow Hub, or train your own model with TensorFlow Lite Model Maker or TensorFlow. Please refer to Custom models with ML Kit to learn more.

iOS Android