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

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

本文例項為大家分享了OpenCV+face++實現實時人臉識別解鎖功能的具體程式碼,供大家參考,具體內容如下

1.背景

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

2.知識儲備

  • python基礎語法
  • opencv
  • face++文件
  • 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,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,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,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,flags=cv2.cv.CV_HAAR_SCALE_IMAGE)
    for x,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"刷臉失敗",win32con.MB_OK)
  #print u"不是同一個人"

執行效果:

因為樣本圖片是我,所以用白巖鬆是不能成功的。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。