from email.mime import image
import imghdr
from turtle import title
import cv2
import matplotlib.pyplot as plt
import numpy as np

def cv_show(img,name):
    cv2.imshow(name,img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()


#模板匹配
def template_match(img,template):
    res=cv2.matchTemplate(img,template,cv2.TM_CCOEFF_NORMED) # 参数为输入图像  模板  参数为匹配方法
    min_val,max_val,min_loc,max_loc=cv2.minMaxLoc(res) # 参数为输入图像  参数为输出最小值  参数为输出最大值  参数为输出最小值的位置  参数为输出最大值的位置
    return max_val,max_loc

img=cv2.imread('lena.jpg')  # 读取图片
template=cv2.imread('lena_part.png') # 读取图片 0表示灰度图 1表示彩色图

max_val,max_loc=template_match(img,template)
top_left=max_loc
bottom_right=(top_left[0]+template.shape[1],top_left[1]+template.shape[0])
cv2.rectangle(img,top_left,bottom_right,(0,0,255),2)
cv_show(img,'after_template_match')
#保存
cv2.imwrite('lena_template_match.jpg',img)