模板匹配找人脸区域再识别口罩

一、先用模板匹配找出人脸区域，然后再匹配口罩

import cv2 as cv
import numpy as np

cap = cv.VideoCapture(0)  #也可以通过1选择第二个摄像头
while True:
    ret, frame = cap.read()  #第一个参数判断是否读取到帧
    frame = cv.flip(frame, 1)  #把图像翻转，可以试试要这一句和不要的区别，参数可以改
    img_gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
    #人脸匹配
    template = cv.imread("E:\\python opencv\\demo2\\face3.jpg", 0)
    w, h = template.shape[::-1]

    res = cv.matchTemplate(img_gray, template, cv.TM_CCOEFF)
    min_val, max_val, min_loc, max_loc = cv.minMaxLoc(res)

    top_left = max_loc
    bottom_right = (top_left[0] + w, top_left[1] + h)

    #人脸区域提取出来
    img1 = frame[top_left[0]:bottom_right[0], top_left[1]:bottom_right[1]]

    cv.rectangle(frame, top_left, bottom_right, 255, 2)
    #口罩匹配
    img_gray1 = cv.cvtColor(img1, cv.COLOR_BGR2GRAY)
    template1 = cv.imread("E:\\python opencv\\demo2\\wiki8.jpg", 0)
    w1, h1 = template1.shape[::-1]
    res1 = cv.matchTemplate(img_gray1, template1, cv.TM_CCOEFF)
    min_val1, max_val1, min_loc1, max_loc1 = cv.minMaxLoc(res1)
    top_left1 = max_loc1
    bottom_right1 = (top_left1[0] + w1, top_left1[1] + h1)
    cv.rectangle(img1, top_left1, bottom_right1, 0, 2)
    frame[top_left[0]:bottom_right[0], top_left[1]:bottom_right[1]] = img1
    cv.imshow('res', frame)

    if cv.waitKey(1) == ord('q'):
        break
cap.release()
cv.destroyAllWindows()

效果不太好，小黑框容易框错位置，而且模板匹配的框的大小只跟输入的模板有关系，应该可以把它放大或缩小效果可能更好一点，还有头移到边缘的时候好像会卡死，没去找原因了。（混一混，混一混）