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