Python 24.opencv 多对象模板匹配

import cv2
import numpy as np

img_rgb = cv2.imread('mali.PNG')
img_rgb_COPY = img_rgb

img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
template = cv2.imread('kcoin.PNG', 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)

res = np.hstack((img_rgb, img_rgb_COPY))

cv2.imshow('img', res)
cv2.waitKey(0)
cv2.destroyAllWindows()