Python cv2模板匹配

时间:2019-03-16 00:52:49

标签: python cv2

我一直在尝试制作一个程序,如果图像中有匹配项,该程序可以执行某项操作,例如函数。但是我找不到一种方法让程序知道是否存在匹配项。

import cv2
import numpy as np
from matplotlib import pyplot as plt

img_rgb = cv2.imread('image.jpg')
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
template = cv2.imread('template.jpg',0)
w, h = template.shape[::-1]

res = cv2.matchTemplate(img_gray,template,cv2.TM_CCOEFF_NORMED)
threshold = 0.8
loc = np.where( res >= threshold)
for pt in zip(*loc[::-1]):
    cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0,0,255), 2)

cv2.imwrite('res.png',img_rgb)

0 个答案:

没有答案