import cv2

cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")

#-*- coding: utf-8 -*-
# import openCV的庫(kù)
import cv2
import os, math, operator
from PIL import Image
from functools import reduce


###調(diào)用電腦攝像頭檢測(cè)人臉并截圖

def CatchPICFromVideo(window_name, path_name):
 cv2.namedWindow(window_name)

 #電腦攝像頭
 cap = cv2.VideoCapture(0)

 #告訴OpenCV使用人臉識(shí)別分類(lèi)器
 classfier = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")

 #檢測(cè)人臉后要畫(huà)的邊框的顏色
 color = (0, 255, 0)

 while cap.isOpened():
 ok, frame = cap.read() #讀取一幀數(shù)據(jù)
 if not ok:
  break

 grey = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) #將當(dāng)前楨圖像轉(zhuǎn)換成灰度圖像

 #人臉檢測(cè)，1.2和2分別為圖片縮放比例和需要檢測(cè)的有效點(diǎn)數(shù)
 faceRects = classfier.detectMultiScale(grey, scaleFactor = 1.2, minNeighbors = 3, minSize = (32, 32))
 if len(faceRects) > 0:  #大于0則檢測(cè)到人臉
  for faceRect in faceRects: #單獨(dú)框出每一張人臉
  x, y, w, h = faceRect

   #畫(huà)出矩形框
  cv2.rectangle(frame, (x - 10, y - 10), (x + w + 10, y + h + 10), color, 2)
  
  k = cv2.waitKey(100) #每0.1秒讀一次鍵盤(pán)
  if k == ord("z") or k == ord("Z"): #如果輸入z
   #將當(dāng)前幀保存為圖片
   img_name = path_name
   print(img_name)
   image = frame[y - 10: y + h + 10, x - 10: x + w + 10]
   cv2.imwrite(img_name, image,[int(cv2.IMWRITE_PNG_COMPRESSION), 9])
   break 
   
 #顯示圖像
 cv2.imshow(window_name, frame)
 #退出攝像頭界面
 c = cv2.waitKey(100)
 if c == ord("q") or c == ord("Q"): 
  break

 #釋放攝像頭并銷(xiāo)毀所有窗口
 cap.release()
 cv2.destroyAllWindows()


os.system("cls") #清屏
recogname = "recogface.jpg" #預(yù)存的人臉文件
CatchPICFromVideo("get face",recogname)