快捷導(dǎo)航

教你如何用Python實現(xiàn)人臉識別（含源代碼）

更新時間：2021年06月22日 17:14:25 作者：python588

Python可以從圖像或視頻中檢測和識別你的臉.人臉檢測與識別是計算機視覺領(lǐng)域的研究熱點之一.人臉識別的應(yīng)用包括人臉解鎖、安全防護等,醫(yī)生和醫(yī)務(wù)人員利用人臉識別來獲取病歷和病史,更好地診斷疾病,需要的朋友可以參考下

工具與圖書館

Python-3.x
CV2-4.5.2
矮胖-1.20.3
人臉識別-1.3.0

若要安裝上述軟件包，請使用以下命令。

pip install numpy opencv-python

要安裝FaceRecognition，首先安裝dlib包。

pip install dlib

現(xiàn)在，使用以下命令安裝面部識別模塊

pip install face_recognition

下載人臉識別Python代碼

請下載python面部識別項目的源代碼： 人臉識別工程代碼

項目數(shù)據(jù)集

我們可以使用我們自己的數(shù)據(jù)集來完成這個人臉識別項目。對于這個項目，讓我們以受歡迎的美國網(wǎng)絡(luò)系列“老友記”為數(shù)據(jù)集。該數(shù)據(jù)集包含在面部識別項目代碼中，您在上一節(jié)中下載了該代碼。

建立人臉識別模型的步驟

在繼續(xù)之前，讓我們知道什么是人臉識別和檢測。

人臉識別是從照片和視頻幀中識別或驗證一個人的臉的過程。

人臉檢測是指在圖像中定位和提取人臉(位置和大小)以供人臉檢測算法使用的過程。

人臉識別方法用于定位圖像中唯一指定的特征。在大多數(shù)情況下，面部圖片已經(jīng)被移除、裁剪、縮放和轉(zhuǎn)換為灰度。人臉識別包括三個步驟：人臉檢測、特征提取、人臉識別。

OpenCV是一個用C++編寫的開源庫.它包含了用于計算機視覺任務(wù)的各種算法和深度神經(jīng)網(wǎng)絡(luò)的實現(xiàn)。

1.準(zhǔn)備數(shù)據(jù)集

創(chuàng)建2個目錄，訓(xùn)練和測試。從互聯(lián)網(wǎng)上為每個演員選擇一個圖片，并下載到我們的“火車”目錄中。確保您所選擇的圖像能夠很好地顯示人臉的特征，以便對分類器進行分類。

為了測試模型，讓我們拍攝一張包含所有強制轉(zhuǎn)換的圖片，并將其放到我們的“test”目錄中。

為了您的舒適，我們增加了培訓(xùn)和測試數(shù)據(jù)與項目代碼。

2.模型的訓(xùn)練

首先導(dǎo)入必要的模塊。

import face_recognition as fr
import cv2
 
import numpy as np
import os

人臉識別庫包含幫助人臉識別過程的各種實用程序的實現(xiàn)。

現(xiàn)在，創(chuàng)建2個列表來存儲圖像(人員)的名稱及其各自的臉編碼。

path = "./train/"
 
known_names = []
known_name_encodings = []
 
images = os.listdir(path)

人臉編碼是一種值的矢量，它代表著臉部特征之間的重要度量，如眼睛之間的距離、額頭的寬度等。

我們循環(huán)遍歷火車目錄中的每個圖像，提取圖像中的人的姓名，計算其臉編碼向量，并將信息存儲在相應(yīng)的列表中。

for _ in images:
image = fr.load_image_file(path + _)
image_path = path + _
encoding = fr.face_encodings(image)[0]
 
known_name_encodings.append(encoding)
known_names.append(os.path.splitext(os.path.basename(image_path))[0].capitalize())

3.在測試數(shù)據(jù)集中測試模型

如前所述，我們的測試數(shù)據(jù)集只包含一個包含所有人員的圖像。

使用CV2 imread()方法讀取測試映像。

test_image = "./test/test.jpg"
 
image = cv2.imread(test_image)

人臉識別庫提供了一種名為Face_Locations()的有用方法，它定位圖像中檢測到的每個人臉的坐標(biāo)(左、下、右、上)。使用這些位置值，我們可以很容易地找到臉編碼。

face_locations = fr.face_locations(image)
 
face_encodings = fr.face_encodings(image, face_locations)

我們循環(huán)遍歷每個面部位置及其在圖像中的編碼。然后，我們將這種編碼與“列車”數(shù)據(jù)集中的人臉編碼進行比較。

然后計算人臉距離，即計算測試圖像編碼和訓(xùn)練圖像編碼之間的相似性。現(xiàn)在，我們從它選取最小值距離，表示測試圖像的這張臉是訓(xùn)練數(shù)據(jù)集中的人之一。

現(xiàn)在，使用CV2模塊中的方法繪制一個帶有面部位置坐標(biāo)的矩形。

for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
   matches = fr.compare_faces(known_name_encodings, face_encoding)
   name = ""
 
   face_distances = fr.face_distance(known_name_encodings, face_encoding)
   best_match = np.argmin(face_distances)
 
   if matches[best_match]:
       name = known_names[best_match]
 
   cv2.rectangle(image, (left, top), (right, bottom), (0, 0, 255), 2)
   cv2.rectangle(image, (left, bottom - 15), (right, bottom), (0, 0, 255), cv2.FILLED)
 
   font = cv2.FONT_HERSHEY_DUPLEX
   cv2.putText(image, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)

使用CV2模塊的imShow()方法顯示圖像。