举例说明在python中用模板匹配识别物体
时间: 2023-01-11 15:55:36 浏览: 110
在 Python 中可以使用模板匹配来识别图像中的物体。模板匹配是一种图像处理技术,它可以在输入图像中查找和模板图像匹配的区域。
下面是一个简单的例子,使用 Python 库 OpenCV 和 NumPy 实现模板匹配:
```
import cv2
import numpy as np
# 读取输入图像和模板图像
input_image = cv2.imread('input.jpg')
template = cv2.imread('template.jpg')
# 将输入图像转换为灰度图像
input_image_gray = cv2.cvtColor(input_image, cv2.COLOR_BGR2GRAY)
# 将模板图像转换为灰度图像
template_gray = cv2.cvtColor(template, cv2.COLOR_BGR2GRAY)
# 进行模板匹配
result = cv2.matchTemplate(input_image_gray, template_gray, cv2.TM_CCOEFF_NORMED)
# 设置阈值,确定匹配的位置
threshold = 0.8
loc = np.where(result >= threshold)
# 在输入图像中绘制矩形框
for pt in zip(*loc[::-1]):
cv2.rectangle(input_image, pt, (pt[0] + w, pt[1] + h), (0, 0, 255), 2)
# 显示输入图像
cv2.imshow('Input image', input_image)
cv2.waitKey(0)
```
在上面的代码中,我们首先读取了输入图像和模板图像,然后将两者都转换为灰度图像。接下来,使用函数 `cv2.matchTemplate()` 在输入图像中查找和模板图像匹配的区域。
阅读全文