python模板匹配代码
时间: 2023-07-23 17:01:17 浏览: 97
以下是使用 OpenCV 库进行模板匹配的 Python 代码示例:
```python
import cv2
# 读取原始图像和模板图像
img = cv2.imread('original_image.png')
template = cv2.imread('template_image.png')
# 获取模板图像的尺寸
h, w = template.shape[:2]
# 使用 TM_CCOEFF_NORMED 方法进行模板匹配
res = cv2.matchTemplate(img, template, cv2.TM_CCOEFF_NORMED)
# 设置匹配阈值
threshold = 0.8
# 获取匹配结果中大于阈值的像素位置
loc = cv2.findNonZero(res > threshold)
# 在原始图像中绘制矩形框标记匹配位置
for pt in loc:
x, y = pt[0]
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 2)
# 显示匹配结果
cv2.imshow('Matching Result', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
其中,`cv2.matchTemplate()` 函数用于进行模板匹配,匹配结果保存在 `res` 中。通过设定阈值 `threshold` 来筛选出匹配结果中大于该阈值的像素位置,最后在原始图像中绘制矩形框标记匹配位置。
相关问题
python opencv 模板匹配代码
模板匹配是一种在图像中查找模板图像位置的方法。OpenCV 提供了 `cv2.matchTemplate()` 函数来实现模板匹配。以下是一个简单的示例代码,展示了如何使用 OpenCV 进行模板匹配:
```python
import cv2
import numpy as np
# 读取主图像和模板图像
main_image = cv2.imread('main_image.jpg', cv2.IMREAD_COLOR)
template = cv2.imread('template.jpg', cv2.IMREAD_COLOR)
# 获取模板图像的尺寸
template_height, template_width = template.shape[:2]
# 执行模板匹配
result = cv2.matchTemplate(main_image, template, cv2.TM_CCOEFF_NORMED)
# 设置匹配阈值
threshold = 0.8
loc = np.where(result >= threshold)
# 在匹配位置绘制矩形框
for pt in zip(*loc[::-1]):
cv2.rectangle(main_image, pt, (pt[0] + template_width, pt[1] + template_height), (0, 0, 255), 2)
# 显示结果图像
cv2.imshow('Detected', main_image)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
### 代码解释
1. **导入库**:
```python
import cv2
import numpy as np
```
导入 OpenCV 和 NumPy 库。
2. **读取图像**:
```python
main_image = cv2.imread('main_image.jpg', cv2.IMREAD_COLOR)
template = cv2.imread('template.jpg', cv2.IMREAD_COLOR)
```
读取主图像和模板图像。
3. **获取模板尺寸**:
```python
template_height, template_width = template.shape[:2]
```
获取模板图像的高度和宽度。
4. **执行模板匹配**:
```python
result = cv2.matchTemplate(main_image, template, cv2.TM_CCOEFF_NORMED)
```
使用 `cv2.matchTemplate()` 函数进行模板匹配。
5. **设置匹配阈值并找到匹配位置**:
```python
threshold = 0.8
loc = np.where(result >= threshold)
```
设置匹配阈值,并找到所有匹配位置。
6. **绘制矩形框**:
```python
for pt in zip(*loc[::-1]):
cv2.rectangle(main_image, pt, (pt[0] + template_width, pt[1] + template_height), (0, 0, 255), 2)
```
在匹配位置绘制红色矩形框。
7. **显示结果图像**:
```python
cv2.imshow('Detected', main_image)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
显示结果图像,并等待用户按键关闭窗口。
python opencv 模板匹配 代码范例
好的,下面提供一下python opencv模板匹配的代码范例:
```python
import cv2
# 读取原图和模板图
img = cv2.imread('original_image.jpg')
template = cv2.imread('template_image.jpg')
# 获取模板图的尺寸
h, w = template.shape[:2]
# 选择匹配算法,这里选择 TM_CCOEFF_NORMED
# https://docs.opencv.org/2.4/modules/imgproc/doc/object_detection.html#cv2.matchTemplate
res = cv2.matchTemplate(img, template, cv2.TM_CCOEFF_NORMED)
# 设置匹配阈值,找到所有可能的匹配位置
threshold = 0.9
locs = cv2.findNonZero(res > threshold)
# 在原图上画出匹配位置的矩形框
for loc in locs:
x, y = loc[0]
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 2)
# 显示匹配结果
cv2.imshow('result', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
希望能够帮到你。
阅读全文