opencv python 识别图片中不同物体的颜色并用矩形框标记并注释文字
时间: 2023-09-17 19:11:25 浏览: 253
下面是一个简单的示例代码,可以实现使用 OpenCV Python 识别图片中不同物体的颜色,并用矩形框标记并注释文字。代码中使用的是 HSV 颜色空间进行颜色识别,你可以根据需要修改颜色阈值来适应不同的场景。
```python
import cv2
# 读取图片
img = cv2.imread('test.jpg')
# 转换为 HSV 颜色空间
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 定义颜色阈值(蓝色)
lower_blue = np.array([100, 50, 50])
upper_blue = np.array([130, 255, 255])
# 对图像进行颜色分割
mask = cv2.inRange(hsv, lower_blue, upper_blue)
# 查找轮廓
contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 标记物体并注释文字
for contour in contours:
# 获取矩形框坐标
x, y, w, h = cv2.boundingRect(contour)
# 绘制矩形框
cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)
# 添加文字注释
cv2.putText(img, 'blue', (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
# 显示结果
cv2.imshow('result', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
在上述代码中,读取了一张名为 `test.jpg` 的图片,将其转换为 HSV 颜色空间,然后定义了蓝色的颜色阈值,对图像进行颜色分割,最后查找轮廓并标记物体并注释文字。你可以根据需要修改颜色阈值的参数来适应不同的场景。
阅读全文