Missed TensorFlow World? Check out the recap. Learn more

tfio.image.draw_bounding_boxes

View source on GitHub

Draw bounding boxes on a batch of images.

tfio.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).