OpenCV+face++實現實時人臉識別解鎖功能
阿新 • • 發佈:2018-12-31
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, 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"不是同一個人"
執行效果:
因為樣本圖片是我,所以用白巖鬆是不能成功的。