在Python中使用draw_bounding_box_on_image_array()函数绘制边界框及颜色标识
发布时间:2023-12-15 19:11:54
在Python中,可以使用TensorFlow提供的draw_bounding_box_on_image_array()函数来绘制边界框及颜色标识带。该函数接受一个图像数组和边界框坐标作为输入,并在图像上绘制相应的边界框及颜色标识带。
以下是一个使用例子,假设我们有一张图像和对应的边界框坐标,我们将使用draw_bounding_box_on_image_array()函数将边界框绘制在图像上:
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
# 图像数组
image_array = np.array([[0.1, 0.5, 0.3],
[0.3, 0.2, 0.8],
[0.4, 0.7, 0.9]])
# 边界框坐标(左上角、右下角)
bounding_box = np.array([[0.2, 0.2, 0.6, 0.6]])
# 绘制边界框及颜色标识带
draw_bounding_box_on_image_array(image_array, bounding_box)
# 显示图像
plt.imshow(image_array, cmap='gray')
plt.show()
运行以上代码,会得到一张包含边界框的图像。
要注意的是,在使用draw_bounding_box_on_image_array()函数之前,需要先安装并导入TensorFlow和Matplotlib库。
