opencv模板匹配查找图像(python)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import cv2
import numpy as np from cv2 import COLOR_BGR2GRAY def main(): # 读取原图 img_rgb = cv2.imread("d:/img-src.png") # 转为灰度图 img_gray = cv2.cvtColor(img_rgb, COLOR_BGR2GRAY) # 读取模版图 template = cv2.imread("d:/img-tmp.png", 0) # 获取模版图宽高 w, h = template.shape[::-1] # 模板匹配 res = cv2.matchTemplate(img_gray, template, cv2.TM_CCOEFF_NORMED) # threshold 和 res_ts 用于阈值设定匹配 #threshold = 0.9 #res_ts = np.where(res >= threshold) min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res) top_left = max_loc bottom_right = (top_left[0] + w, top_left[1] + h) # 画方框，[0,0,255] 颜色，2 线宽 cv2.rectangle(img_rgb, top_left, bottom_right, [0,0,255], 2) cv2.imwrite("d:/res.png",img_rgb) img_out = cv2.imread("d:/res.png") cv2.imshow("123", img_out) cv2.waitKey(0) cv2.destroyAllWindows() if __name__ == "__main__": main() 

 来源： 莆田SEO