1. 程式人生 > >OpenCV+face++實現實時人臉識別解鎖功能

OpenCV+face++實現實時人臉識別解鎖功能

1.背景

最近做一個小東西,需要登入功能,一開始做的就是普通的密碼登入功能,但是之前看到過python可以做人臉識別,所以我就開了下腦洞,能不能實現一個自己的刷臉解鎖功能。

2.知識儲備

  1. python基礎語法
  2. opencv
  3. face++文件
  4. requests庫

3.基本思路

準備一張你想要被識別出的人臉照片,後面刷臉就是按照這張照片來識別,如果和照片中是同一個人就解鎖,刷臉就是開啟攝像頭獲取電腦面前人的人臉,然後與之前那張照片比對。

4.程式碼講解

看下識別的效果:
這裡寫圖片描述

第一個函式就是開啟攝像頭並儲存圖片:

#從攝像頭讀取圖片並儲存
def getpicture
():
cap = cv2.VideoCapture(0)#開啟攝像頭 cascade = cv2.CascadeClassifier("E:\OpenCV\sources\data\haarcascades\haarcascade_frontalface_default.xml")#這裡是是自己的人臉識別xml路徑 while True: # get a frame ret, frame = cap.read()#捕獲圖片 # show a frame gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)#轉為灰度圖
rect = cascade.detectMultiScale(gray, scaleFactor=1.15, minNeighbors=5, minSize=(5, 5),flags=cv2.cv.CV_HAAR_SCALE_IMAGE) # 使用模板匹配圖形 for x, y, z, w in rect: cv2.rectangle(frame, (x, y), (x + z, y + w), (0, 0, 255), 2)# 函式的引數分別為:影象,左上角座標,右下角座標,顏色,寬度 cv2.imshow("capture"
, frame) if cv2.waitKey(1) & 0xFF == ord('q'):#按下q拍照 cv2.imwrite("images\client.jpg", frame)#相對路徑,儲存圖片 break cap.release() cv2.destroyAllWindows()

第二個函式是將樣本圖片與攝像頭讀取的圖片上傳到face++進行處理,並拿到它的face_token,該函式主要用到的就是requests庫與face++的api。

def upload_img(fileDir, oneface=True):
    url = '%s/detect?api_key=%s&api_secret=%s' % (
        BASE_URL, API_KEY, API_SECRET)
    #注意引數名與api文件一致
    files = {'image_file': (os.path.basename(fileDir), open(fileDir, 'rb'),
                     mimetypes.guess_type(fileDir)[0]), }
    r = requests.post(url, files=files)
    faces = r.json().get('faces')
    #print faces
    if faces is None:
        print('There is no face found in %s' % fileDir)
    else:
        return faces[0]['face_token']#返回face_token

第三個函式是比較兩張圖片的face_token:

def compare(face_token1,face_token2):
    url = '%s/compare' % BASE_URL
    params = BASE_PARAMS
    params['face_token1'] = face_token1
    params['face_token2'] = face_token2
    r = requests.post(url, params)
    #print r.status_code
    #print r.json()
    return r.json().get('confidence')#返回兩張照片的相似度

最後判斷一下compare()函式的返回值就知道兩張圖片是不是同一個人了,再程式中加一個判斷語句就可以實現基本的解鎖功能了。

完整程式碼:

#! usr/bin/env python
#-*- coding:utf-8 -*-

import requests
import os
import mimetypes    #判斷檔案型別
import cv2
import time
import win32api
import win32con

BASE_URL = "https://api-cn.faceplusplus.com/facepp/v3"
API_KEY = "g_vhMthXCQEzF0gZG5-o0ICNDhr3-80b"
API_SECRET = "2HD5ysubTeZTwo20JJTudY0cvZN1BPLt"
BASE_PARAMS = {
    'api_key':'g_vhMthXCQEzF0gZG5-o0ICNDhr3-80b',
    'api_secret':'2HD5ysubTeZTwo20JJTudY0cvZN1BPLt'
}
def upload_img(fileDir, oneface=True):
    url = '%s/detect?api_key=%s&api_secret=%s' % (
        BASE_URL, API_KEY, API_SECRET)
    #注意引數名與api文件一致
    files = {'image_file': (os.path.basename(fileDir), open(fileDir, 'rb'),
                     mimetypes.guess_type(fileDir)[0]), }
    r = requests.post(url, files=files)
    faces = r.json().get('faces')
    #print faces
    if faces is None:
        print('There is no face found in %s' % fileDir)
    else:
        return faces[0]['face_token']



def compare(face_token1,face_token2):
    url = '%s/compare' % BASE_URL
    params = BASE_PARAMS
    params['face_token1'] = face_token1
    params['face_token2'] = face_token2
    r = requests.post(url, params)
    #print r.status_code
    #print r.json()
    return r.json().get('confidence')


def getpicture():
    cap = cv2.VideoCapture(0)
    cascade = cv2.CascadeClassifier("E:\OpenCV\sources\data\haarcascades\haarcascade_frontalface_default.xml")#這裡是是自己的人臉識別xml路徑
    while True:
        # get a frame
        ret, frame = cap.read()
        # show a frame
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        rect = cascade.detectMultiScale(gray, scaleFactor=1.15, minNeighbors=5, minSize=(5, 5),flags=cv2.cv.CV_HAAR_SCALE_IMAGE)
        for x, y, z, w in rect:
            cv2.rectangle(frame, (x, y), (x + z, y + w), (0, 0, 255), 2)
        cv2.imshow("capture", frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            cv2.imwrite("images\client.jpg", frame)#相對路徑
            break
    cap.release()
    cv2.destroyAllWindows()


getpicture()
print u"                                     資料讀取中。。。。\n"
face1 = upload_img(u"images\demo4.jpg")
print u"                                               正在校對人臉。。。。。\n"
time.sleep(5)#防止出現qps
print u"                                                             再等一下。。。。。\n"
face2 = upload_img(u"images\client.jpg")
confidence = compare(face1,face2)
if confidence>=70:
    #print u"同一個人"
    #win32api.ShellExecute(0,'op','genealogy.exe','','',1)
    win32api.MessageBox(0, u"刷臉成功", u"家譜管理系統", win32con.MB_OK)
    #這裡寫你想要繼續執行的程式碼
else:
    win32api.MessageBox(0, u"刷臉失敗", u"家譜管理系統", win32con.MB_OK)
    #print u"不是同一個人"

執行效果:
failed
因為樣本圖片是我,所以用白巖鬆是不能成功的。