用Python写一个简单的脸部识别项目(1)|opencv 截取脸部图片

这次先介绍怎么从摄像头抽取头像

从摄像头切取照片

目标

• 从摄像头切取20张同一人的脸部照片
  • 识别出人脸所在的位置
  • 截取并保存

getImageFromCamera.py

# -*- coding: utf-8 -*-
import cv2
import os

cap = cv2.VideoCapture(0)
faceDetector=cv2.CascadeClassifier('haarcascade_frontalface_default.xml')

if not os.path.exists('dataSet'):
    os.makedirs('dataSet')

id,name=input('スペースで区切って，IDと名前を入力してください: ').split()

picnumber =0
while True:
    ret,img=cap.read()
    gray =cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    # size, likelyhood
    faces=faceDetector.detectMultiScale(gray,1.3,5)
    for (x,y,w,h) in faces:
        # draw rectangle around faces (x,y) start point w=width h=high ,color , thickness
        cv2.rectangle(img, (x,y),(x+w,y+h),(255,0,0),2)
        roi_gray=gray[y:y+h,x:x+w]
        roi_color=img[y:y+h,x:x+w]

        cv2.imwrite("dataSet/User."+str(id)+"."+name+"."+str(picnumber)+".jpg",gray[y:y+h,x:x+w])
        picnumber+=1
        cv2.waitKey(1000)

    cv2.imshow('img',img)
                        #take 20 pics
    if cv2.waitKey(1) & picnumber==20:
        break

cap.release()
cv2.destroyAllWindows()

1. 读取摄像头的影像
2. 输入用户ID和用户名，新建一个文件夹储存照片
3. while文：不停扫描画面，如果发现人脸，截取图片保存至文件夹中，文件命名参照相应的用户名和ID
4. 截取完20张关闭摄像头

今天先写到这，有问题请留言。