python 封裝dlib模型進行人臉識別系統的登錄認證

1、直接上干貨

#!/usr/bin/python
# -*- coding: utf-8 -*-
import time
import dlib
import numpy as np
class faceDiscernModel:
    def __init__(self):
        # 加載預訓練人臉檢測CNN模型
        self.cnn_face_model = "./model/mmod_human_face_detector.dat"
        self.cnn_face_detector = dlib.cnn_face_detection_model_v1(self.cnn_face_model)

        # 加載人臉特征檢測模型
        self.predictor_path = "./model/shape_predictor_5_face_landmarks.dat"
        self.predictor = dlib.shape_predictor(self.predictor_path)

        # 加載人臉特征提取模型
        self.featureModel = "./model/dlib_face_recognition_resnet_model_v1.dat"
        self.feater = dlib.face_recognition_model_v1(self.featureModel)

    def get_faces_feature(self,imgPath):

        # 讀取人臉圖片
        img = dlib.load_rgb_image(imgPath)

        feat = []

        # 檢測每個人臉的邊界框
        dets = self.cnn_face_detector(img, 1)

        # len(dets) 是檢測到的人臉數量
        for i, d in enumerate(dets):
            # print("Detection {}: Left: {} Top: {} Right: {} Bottom: {} Confidence: {}".format(
            #     i, d.rect.left(), d.rect.top(), d.rect.right(), d.rect.bottom(), d.confidence))

            # 檢測 box i 內的人臉關鍵點
            shape = self.predictor(img, d.rect)

            # 計算特征向量
            face_descriptor = self.feater.compute_face_descriptor(img, shape)

            feat.append(face_descriptor)

        return feat

    def Euclidean_distance_test(self,feature_1,feature_2):
        return np.sqrt(np.sum((np.array(feature_1)-np.array(feature_2))**2))

    def face_compare(self,imgPath_1,imgPath_2,assess=0.6):
        feature_1 = self.get_faces_feature(imgPath_1)
        feature_2 = self.get_faces_feature(imgPath_2)
        score = self.Euclidean_distance_test(feature_1=feature_1,feature_2=feature_2)
        if score > assess:
            return False
        elif 0< score < assess:
            return True
        else:
            return False


if __name__ == '__main__':

    imgPath_1 = "./faceImage/test.jpg"
    imgPath_2 = "./faceImage/test1.jpg"
    faceDiscern = faceDiscernModel()
    st = time.time()
    result = faceDiscern.face_compare(imgPath_1,imgPath_2)

    print(time.time()-st)
    print(result)

　　

2、模型下載地址

　　http://dlib.net/files/