Python-OpenCV 图像处理（十二）：模板匹配

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

__author__ = "zxsuperstar"
__email__ = "zxsuperstar@163.com"

"""
模板匹配
"""
def template_demo():
    tpl = cv2.imread("mb.jpg")
    target = cv2.imread("t.jpg")
    cv2.imshow("template",tpl)
    cv2.imshow("target",target)
    methods = [cv2.TM_SQDIFF_NORMED,cv2.TM_CCORR_NORMED,cv2.TM_CCOEFF_NORMED]
    th, tw = tpl.shape[0:2]
    for md in methods:
        print(md)
        results = cv2.matchTemplate(target,tpl,md)
        min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(results)
        if md == cv2.TM_SQDIFF_NORMED:
            tl = min_loc
        else:
            tl = max_loc
        br = (tl[0] + tw ,tl[1]+th)
        cv2.rectangle(target,tl,br,(0,0,255),2)
        cv2.imshow("match-" + np.str(md),target)


if __name__ == "__main__":
    # src = cv2.imread("t.jpg") #blue green red
    # # cv2.namedWindow("image", cv2.WINDOW_AUTOSIZE)
    # cv2.imshow("image",src)