import cv2

def load_img(path,name,mun = 100,add_with = 0):
 # 獲取人臉識別模型
 # 
 # 
 #以下路徑需要更改為自己環(huán)境下xml文件
 #一般在環(huán)境下的Liba\site-packages\cv2\data\haarcascade_frontalface_alt2.xml
 classfier = cv2.CascadeClassifier('F:\\pyhton\\pytonApp\\Lib\\site-packages\\cv2\\data\\haarcascade_frontalface_alt2.xml')
 #
 #
 # 創(chuàng)建一個窗口
 cv2.namedWindow('face')
 # 打開第一個個攝像頭
 cap = cv2.VideoCapture(0)

 i = 0 # 計數(shù)
 if cap.isOpened():
  while i < mun:
   ok,frame = cap.read() # 讀取一幀圖片
   if not ok:
    continue

   faces = classfier.detectMultiScale(frame,1.2,3,minSize=(32,32))

   if len(faces) > 0:
    for face in faces:
     x, y, w, h = face
     cv2.rectangle(frame,(x-add_with,y-add_with), (x+w+add_with,y+h+add_with), (0,255,0), 2)
     img = frame[y-add_with:y+h+add_with,x-add_with:x+w+add_with]
     save_path = path+name+'_'+str(i)+'.jpg'
     print(save_path)
     cv2.imwrite(save_path,img)
     i += 1

   cv2.imshow('face', frame)
   c = cv2.waitKey(10)
   if c & 0xFF == ord('q'):
    break

  cap.release()
  cv2.destroyAllWindows()

if __name__ == '__main__' :
 # 第一個參數(shù)為保存圖片的路徑
 # 第二個參數(shù)為保存圖片名字的開頭
 # 第三個參數(shù)為圖片的數(shù)量
 # 第四個參數(shù)可以調(diào)節(jié)圖片的大小
 load_img('E:\\Screenshots\\home\\','rongdang',1000,20)