人脸识别（一）合照中的人脸识别：python+ face_recognition

先上照片：（原照+识别后）


代码如下：

import numpy as np
import cv2
import face_recognition
from PIL import Image, ImageDraw, ImageFont

# 计算图片的对应编码
huang_xiao_ming = face_recognition.face_encodings(face_recognition.load_image_file('./pic/huang_xiao_ming.jpg'))[0]
deng_chao = face_recognition.face_encodings(face_recognition.load_image_file('./pic/deng_chao.jpg'))[0]
shen_teng = face_recognition.face_encodings(face_recognition.load_image_file('./pic/shen_teng.jpg'))[0]
biden = face_recognition.face_encodings(face_recognition.load_image_file('./pic/biden.jpeg'))[0]
feng_shao_feng = face_recognition.face_encodings(face_recognition.load_image_file('./pic/feng_shao_feng.jpg'))[0]
zheng_kai = face_recognition.face_encodings(face_recognition.load_image_file('./pic/zheng_kai.jpg'))[0]
li_yi_feng = face_recognition.face_encodings(face_recognition.load_image_file('./pic/li_yi_feng.jpg'))[0]
han_geng = face_recognition.face_encodings(face_recognition.load_image_file('./pic/han_geng.jpg'))[0]
tong_da_wei = face_recognition.face_encodings(face_recognition.load_image_file('./pic/tong_da_wei.jpg'))[0]
tong_li_ya = face_recognition.face_encodings(face_recognition.load_image_file('./pic/tong_li_ya.jpg'))[0]
di_li_re_ba = face_recognition.face_encodings(face_recognition.load_image_file('./pic/di_li_re_ba.jpg'))[0]
yang_mi = face_recognition.face_encodings(face_recognition.load_image_file('./pic/yang_mi.jpg'))[0]

known_encodings = [
    huang_xiao_ming,
    deng_chao,
    shen_teng,
    biden,
    feng_shao_feng,
    zheng_kai,
    li_yi_feng,
    han_geng,
    tong_da_wei,
    tong_li_ya,
    di_li_re_ba,
    yang_mi
]

people_names = [ 'Huang Xiao-ming','Deng Chao','Shen teng','Joe Biden','Feng Shao-feng','zheng_kai','li_yi_feng','han_geng','tong_da_wei','tong_li_ya','di_li_re_ba','yang_mi']
people_names_zh = ['黄晓明','邓超','沈腾','拜登','冯绍峰','郑凯','李易峰','韩庚','佟大为','佟丽娅','迪丽热巴','杨幂']


# 将中文显示在图片上 
def cv2ImgAddText(img, text, left, top, textColor=(0, 255, 0), textSize=20):
    if (isinstance(img, np.ndarray)):  #判断是否OpenCV图片类型
        img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    draw = ImageDraw.Draw(img)
    fontText = ImageFont.truetype( 
        "C:/Windows/Fonts/simsun.ttc", textSize, encoding="utf-8")
    draw.text((left, top), text, textColor, font=fontText)
    return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)

test_image = face_recognition.load_image_file("./pic/hezhao.jpg")
test_image_encoding = face_recognition.face_encodings(test_image)[0]
test_image_location = face_recognition.face_locations(test_image)

#(A，B,C,D)对应(top,right,bottom,left)
print('图片中检测到',len(test_image_location),'张脸')

saveImg = test_image
for i in range(0,len(test_image_location)):
    top = test_image_location[i][0]
    right = test_image_location[i][1]
    bottom = test_image_location[i][2]
    left = test_image_location[i][3]
    cv2.rectangle(saveImg,(left,bottom),(right,top),(0,255,0),2)
    # 得到差异度列表，值最小的最相似
    face_image =test_image[top:bottom, left:right]
    face_image_encoding = face_recognition.face_encodings(face_image)
    if len(face_image_encoding):
        distance_list = face_recognition.face_distance(known_encodings,face_image_encoding[0])
        idx=np.argmin(distance_list)
        print(min(distance_list),'\t',people_names_zh[idx])
        if min(distance_list)<0.6:
            saveImg = cv2ImgAddText(saveImg, people_names_zh[idx], left+10, top-20, (255, 0, 0), 20)
        else:
            saveImg = cv2ImgAddText(saveImg,'unknown', left+10, top-20, (255, 0, 0), 20)
    else:
       saveImg = cv2ImgAddText(saveImg, 'none', left, top-20, (255, 0, 0), 20) 


# 注意openCV的图片是BGR模式
cv2.imshow('face-compare',cv2.cvtColor(saveImg, cv2.COLOR_RGB2BGR))

cv2.waitKey(0)
cv2.destroyAllWindows()

几点说明：

（1）openCV不支持显示中文，代码中cv2ImgAddText函数用于添加中文显示

（2）openCV打开及显示的图片是BGR格式的，需要做RGB2BGR的转换

（3）人脸相似度可以用 1-min(distance_list)百分数来计算