應用dlib庫中的face_recognition進行人臉檢測python代碼實現

一.代碼實現

如果你已經完全安裝好dlib和face_recognition,那麼直接在pycharm中運行下面的代碼即可實現圖片中的人臉檢測.值得注意的是,要修改cv2.imread("")中的圖片路徑爲自己的圖片的存儲路徑.

本人使用的是Ubuntu系統.

face_recognition是基於Dlib的深度學習人臉識別庫,在戶外臉部識別檢測數據庫基準(Labeled Faces in the Wild benchmark , LFW)上的準確率達到99.38%.

LFW數據庫鏈接:http://vis-www.cs.umass.edu/lfw/

import dlib
import cv2
import sys
import os
import glob
import numpy as np
import face_recognition
img=cv2.imread("/home/kd/PycharmProjects/python-pro/datasets/4.jpeg")
face_locations=face_recognition.face_locations(img)

#獲取人臉的位置信息
img_faces=[]
for i in range(len(face_locations)):
    x=face_locations[i][3]
    y=face_locations[i][0]
    h=face_locations[i][2]-face_locations[i][0]
    w=face_locations[i][1]-face_locations[i][3]
    img_faces.append([x,y,w,h])
    cv2.rectangle(img,(img_faces[i][0],img_faces[i][1]),(img_faces[i][0]+img_faces[i][2],img_faces[i][1]+img_faces[i][3]),(0,0,255))
cv2.imshow("pic",img)
cv2.waitKey(0)

 二.face_recogntion在Ubuntu中的安裝

1.安裝dlib

# 安裝 Dlib
# 激活 conda 虛擬環境

conda activate tensorflow
# 安裝 Dlib

conda install -c menpo dlib

2.安裝cmake

sudo apt-get install cmake

3.安裝face_recognition

# 激活 conda 虛擬環境
conda activate tensorflow

# 由於 face_recognition 在 conda 中沒有相應的軟件包，因此通過 pip 安裝
pip install face_recognition