總結幾個簡單好用的Python人臉辨識演算法

WBOY
發布: 2023-04-12 14:34:09
轉載
1830 人瀏覽過

哈嘍,大家好。

今天要為大家總結幾個簡單、好用的人臉辨識演算法。人臉辨識是電腦視覺中比較常見的技術,生活中,我們接觸最多的人臉辨識場景是人臉考勤,我之前還專門寫過一篇人臉考勤的項目,有興趣的朋友可以看看。

人臉辨識的演算法最核心的工作是從一張圖片中辨識出人臉的位置。辨識的演算法可以說是多種多樣, 下面我就來為大家一一介紹下。

1. HoG人臉偵測

此演算法採用傳統的機器學習演算法來辨識人臉。傳統機器學習演算法的特點是人工構造特徵,然後將構造好的特徵送入模型訓練。

演算法以HoG擷取圖片中人臉特徵,以SVM演算法進行分類。

HoG(Histogram of Oriented Gradient, 方向梯度直方圖)特徵是一種在電腦視覺和影像處理中用來進行物體偵測的特徵描述子,透過計算和統計影像局部區域的梯度方向直方圖來構成特徵。

dlib庫中有該演算法的實現,下面我們看看核心程式碼

import dlib
# 加载预训练的 HoG 人脸检测器
hog_face_detector = dlib.get_frontal_face_detector()

# 对總結幾個簡單好用的Python人臉辨識演算法进行人脸检测
results = hog_face_detector(imgRGB, 0)

for bbox in results:
x1 = bbox.left()# 人脸左上角x坐标
y1 = bbox.top()# 人脸左上角y坐标
x2 = bbox.right()# 人脸右下角x坐标
y2 = bbox.bottom()# 人脸右下角y坐标
登入後複製

results 存放一張圖中偵測出來的多個人臉, 遍歷results可以得到每張人臉的矩形框。

偵測範例如下:

總結幾個簡單好用的Python人臉辨識演算法

綠色框框出來的是演算法偵測出來的人臉。

HoG 人臉偵測由於採用傳統機器學習演算法,所以效能比較高,在CPU上運作也可以比較快。但它無法偵測小於 80*80 的人臉,對旋轉人臉、非正面人臉,辨識效果也不太好。

2. 深度學習人臉偵測

雖然傳統機器學習演算法偵測更快,但準確度卻有待提升。基於深度學習的人臉偵測演算法往往會更加準確。

這裡介紹的是使用殘差網路ResNet-10透過網路(模型)在影像的單一通道( Single Shot Detector,SSD)中偵測多個人臉。簡稱SSD演算法。

首先,需要將原始圖片進行blob預處理,然後直接送入模型​​,進行檢測

cv2庫提供了演算法的實現,核心程式碼如下:

import cv2

# 加载预训练的 SSD 模型
opencv_dnn_model = cv2.dnn.readNetFromCaffe(
prototxt="models/deploy.prototxt"
, caffeModel="models/res10_300x300_ssd_iter_140000_fp16.caffemodel")

# 原始總結幾個簡單好用的Python人臉辨識演算法 blob 处理
preprocessed_image = cv2.dnn.blobFromImage(image, scalefactor=1.0, size=(300, 300), mean=(104.0, 117.0, 123.0), swapRB=False, crop=False)

# blob 總結幾個簡單好用的Python人臉辨識演算法送入模型
opencv_dnn_model.setInput(preprocessed_image)

# 模型推理,进行人脸检测
results = opencv_dnn_model.forward()

# 遍历人脸
for face in results[0][0]:
# 置信度
face_confidence = face[2]

# 人脸边框的左上角和右下角坐标点
x1 = int(bbox[0] * image_width)
y1 = int(bbox[1] * image_height)
x2 = int(bbox[2] * image_width)
y2 = int(bbox[3] * image_height)
登入後複製

results[0][0]存放了檢測出來的多張人臉,每張人臉用數組表達,數組的第3位存放置信度,可以透過閾值過濾不置信的人臉。陣列的第4~7位元存放偵測出來的人臉矩形框左上角和右下角的座標。

相較於 HoG 人臉偵測,SSD 演算法對遮蔽、非正面人臉也能偵測出來。

總結幾個簡單好用的Python人臉辨識演算法

3. 卷積神經網路人臉偵測

卷積就不多說了,了解電腦視覺的都知道。

dlib函式庫提供了卷積神經網路人臉偵測演算法的實現,用法跟之前類似

import dlib

# 记载预训练模型
cnn_face_detector = dlib.cnn_face_detection_model_v1("models/mmod_human_face_detector.dat")

# 人脸检测
results = cnn_face_detector(imgRGB, 0)

# 遍历每张人脸
for face in results:
# 人脸边框
bbox = face.rect

# 人脸边框的左上角和右下角坐标点
x1 = int(bbox.left() * (width/new_width))
y1 = int(bbox.top() * (height/new_height))
x2 = int(bbox.right() * (width/new_width))
y2 = int(bbox.bottom() * (height/new_height))
登入後複製

results的解析跟上面類似,這裡就不在贅述了。

採用卷積神經網路的人臉偵測演算法優勢很明顯,比前兩個更準確和健壯,也能夠偵測遮擋下的人臉。

總結幾個簡單好用的Python人臉辨識演算法

即便非正面、且光線暗的圖片,也能很好地偵測出來

總結幾個簡單好用的Python人臉辨識演算法

##但此演算法對應的缺點也很明顯,檢測過程所花費的時間非常長,無法在CPU 上即時運作。

4. BlazeFace

上面的演算法要麼精度高、速度慢,要麼速度快,精度低。那有沒有一種檢測演算法,既有高準確率,又有高效能呢?

答案是肯定的,BlazeFace是一種非常輕量級且高度準確的人臉偵測器,號稱亞毫秒級的人臉偵測器。其靈感來自 Single Shot MultiBox Detector (SSD) 和 MobileNetv2。

Mediapipe庫提供了演算法的實現,核心程式碼如下:

import mediapipe as mp

# 画图工具
mp_drawing = mp.solutions.drawing_utils

# 初始化人脸检测模型
mp_face_detection = mp.solutions.face_detection
mp_face_detector = mp_face_detection.FaceDetection(min_detection_confidence=0.4)

results = mp_face_detector.process(imgRGB)

if results.detections:

# 变脸检测出的人脸
for face_no, face in enumerate(results.detections):

# 画人脸关键点
mp_drawing.draw_detection(image=output_image, detection=face, keypoint_drawing_spec=mp_drawing.DrawingSpec(color=(0,255,0),thickness=-1, circle_radius=image_width//115), bbox_drawing_spec=mp_drawing.DrawingSpec(color=(0,255,0),thickness=image_width//180))

# 画人脸框
face_bbox = face.location_data.relative_bounding_box
x1 = int(face_bbox.xmin*image_width)
y1 = int(face_bbox.ymin*image_height)

cv2.rectangle(output_image, pt1=(x1, y1-image_width//20), pt2=(x1+image_width//16, y1), color=(0, 255, 0), thickness=-1)
登入後複製
效果如下:

總結幾個簡單好用的Python人臉辨識演算法

可以看到,BlazeFace演算法不光能偵測人臉,還能辨識出人臉6個關鍵點(眼睛、鼻子、耳朵、嘴巴)。

上面就是今天分享的4個人臉辨識的演算法。

辨識出人臉,我們再做人臉考勤就非常簡單了,把人臉Embedding成向量,計算向量的之間的距離即可。

以上是總結幾個簡單好用的Python人臉辨識演算法的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:51cto.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板
關於我們 免責聲明 Sitemap
PHP中文網:公益線上PHP培訓,幫助PHP學習者快速成長!