匹配多个对象

import cv2
import numpy as np

# 原图
origin = cv2.imread('./image/many-cat.png')
# 原图灰度化
origin_gray= cv2.cvtColor(origin,cv2.COLOR_BGR2GRAY)
# 直接读取模版灰度图
template = cv2.imread('./image/min-cat.png',0)
# 202,167
print(template.shape)
# 1是宽0是高
# w = template.shape[1]
# h = template.shape[0]
h,w = template.shape

# 模版匹配
res = cv2.matchTemplate(origin_gray,template,cv2.TM_CCOEFF_NORMED)
# 设定阈值，认为相关系数大于0.8，就认为是匹配上了
threshold = 0.98
# print(res)
# print(res>=threshold)
# where函数找到位置坐标，是一个元组，一个x轴，一个y轴
# print(np.where(res>=threshold))
# 找到复核条件的坐标，返回x，y的列表，在一起
# print(np.argwhere(res>=threshold))
loc = np.argwhere(res>=threshold)
# print(loc)
for pt in loc:
    bottom_right = (pt[1] + w,pt[0] + h)
    cv2.rectangle(origin,pt[::-1],bottom_right,(0,0,255),2)

# loc = np.where(res >= threshold)
# for pt in zip(*loc[::-1]):
#     bottom_right=(pt[0] + w,pt[1] + h)
#     cv2.rectangle(origin,pt,bottom_right,(0,0,255),2)

cv2.imshow('origin',origin)
cv2.waitKey(0)
cv2.destroyAllWindows()