Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfio.experimental.image.draw_bounding_boxes

View source on GitHub

Draw bounding boxes on a batch of images.

tfio.experimental.image.draw_bounding_boxes(
    images, boxes, texts=None, colors=None, name=None
)

Args:

  • images: A Tensor. Must be one of the following types: float32, half. 4-D with shape [batch, height, width, depth]. A batch of images.
  • boxes: A Tensor of type float32. 3-D with shape [batch, num_bounding_boxes, 4] containing bounding boxes.
  • name: A name for the operation (optional).

Returns:

A Tensor of type uint8 and shape of [height, width, 4] (RGBA).