《python+opencv3.3视频教学 基础入门》模板匹配 笔记

源码：

import cv2 as cv
import numpy as np


def template_demo():
    tpl = cv.imread("C:/cvtest/0002.jpg")
    target = cv.imread("C:/cvtest/0000.jpg")

    cv.imshow("template image", tpl)
    cv.imshow("target image", target)

    # 要使用的相关性因子
    methods = [cv.TM_SQDIFF_NORMED, cv.TM_CCORR_NORMED, cv.TM_CCOEFF_NORMED]
    th, tw = tpl.shape[:2]

    for md in methods:
        print(md)
        # result里面存放的是由相关因子的计算公式所计算出来的值，
        # 每一个像素点都对应一个值
        result = cv.matchTemplate(target, tpl, md)
        min_val, max_val, min_loc, max_loc = cv.minMaxLoc(result)
        if md == cv.TM_SQDIFF_NORMED:
            tl = min_loc
        else:
            tl = max_loc
        br = (tl[0] + tw, tl[1] + th)
        cv.rectangle(target, tl, br, (0, 0, 255), 2)
        cv.imshow("match-" + np.str(md), target)
        # cv.imshow("match-" + np.str(md), reslut)


print("-----------Python OpenCV Tutorial--------------")
src = cv.imread("C:/cvtest/0000.jpg")
cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)

template_demo()

cv.imshow("input image", src)
cv.waitKey(0)

cv.destroyAllWindows()