在图像 Python cv2 上绘制边界框 - Python

简介

在对图像进行处理时，有时需要在图像中标记出某些区域，常见的方式是使用边界框。在 Python 中，可以使用 OpenCV 库中的 cv2.rectangle() 函数来绘制边界框。

示例代码

import cv2
 
# 加载图像
img = cv2.imread('image.jpg')
 
# 定义边界框的顶点坐标
start_point = (100, 100)
end_point = (400, 400)
 
# 定义边框的颜色和厚度
color = (0, 255, 0)
thickness = 3
 
# 使用 cv2.rectangle() 函数绘制边界框
img = cv2.rectangle(img, start_point, end_point, color, thickness)

# 显示图像
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

代码解析

1. 导入 cv2 库；
2. 使用 cv2.imread() 函数加载图像；
3. 定义边界框的顶点坐标，即起点和终点；
4. 定义边框的颜色和厚度；
5. 使用 cv2.rectangle() 函数绘制边界框；
6. 使用 cv2.imshow() 函数显示图像；
7. 使用 cv2.waitKey() 函数等待用户按下任意键，随后使用 cv2.destroyAllWindows() 函数关闭所有窗口。

总结

本文介绍了在 Python 中使用 OpenCV 库中的 cv2.rectangle() 函数绘制边界框的方法，具体实现过程较为简单，需要注意的是要定义好边界框的顶点坐标、颜色和厚度。