前天听人吹牛,说某某的系统好牛,可以识别客户的头像,好高科技巴拉巴拉。
于是昨天找了一下,发现Python的face_regcognition包可以做,代码封装的非常简单,一点识别原理都不懂的人,也可以使用。
face_regcognition 可以实现的功能包括:
在图像上查找所有人脸的位置:
将一张人脸图片与其它图片对比,找出最相似的:比如找一张女星相片是不是范爷的
查找人脸中的嘴唇、下巴等位置。
其它还有支持命令行操作,使用GPU做批量处理加速等等。
使用环境:Python3.6 + Windows10
因为依赖dlib库的原因,在安装前需要先安装dlib。建议使用dlib的官方包安装,否则容易报错
dlib地址
一、在示例图片上找到人脸的位置,并做框线标注
示例图片
test03.jpg
标注后的图片效果
test_result.jpg
示例
import face_recognition as frimport cv2 img = fr.load_image_file("tmp/locations.jpg") # 加载图片保存为numpy数组loc = fr.face_locations(img) # 查找人脸位置,返回位置坐标的list, 真的只有一行哦# 遍历坐标list,给每个人脸位置画上标线框for i in loc: cv2.rectangle(img, (i[3],i[0]), (i[1],i[2]), color=(0,255,255), thickness=3)# 保存标后的图片cv2.imwrite('/tmp/test_result.jpg', img)
二、查找最相似的人脸
示例图片:在这四张相片中,找到与目标图片最相似的人脸。范爷,范爷她弟,大景甜,明日花绮罗(好像混进了不一样的东西)
fanbingbing.jpg
fanchenchen.jpg
jingtian.jpg
mingrihuayiluo.jpg
目标图片
test01.jpg
示例
import face_recognition as fr# 读取图像img_a = fr.load_image_file("tmp/fanbingbing.jpg") img_b = fr.load_image_file('tmp/fanchenchen.jpg') img_c = fr.load_image_file('tmp/jingtian.jpg') img_d = fr.load_image_file('tmp/mingrihuayiluo.jpg') img_x = fr.load_image_file('tmp/test01.jpg')# 进行特征编码encoding_a = fr.face_encodings(img_a)[0] encoding_b = fr.face_encodings(img_b)[0] encoding_c = fr.face_encodings(img_c)[0] encoding_d = fr.face_encodings(img_d)[0] encoding_x = fr.face_encodings(img_x)[0]# 将示例图片与目标人脸逐一对比,返回 list, 包含True/False, 表示是否匹配fr.compare_faces([encoding_a, encoding_b, encoding_c, encoding_d], encoding_x)# 返回 [True, True, True, False]# 我擦,不对啊,怎么第2,3张也是True。# 差值默认为0.5,越小对比越严格fr.compare_faces([encoding_a, encoding_b, encoding_c, encoding_d], encoding_x, tolerance=0.4)# 返回值[True, False, False, False],得知目标人脸属于范爷
三、对比示例人脸与目标人脸的相似度
将上面示例的人脸与目标人脸对比,计算欧氏距离。距离越小越相似。
fr.face_distance([encoding_a, encoding_b, encoding_c, encoding_d], encoding_x) # 返回array([ 0.33231199, 0.59964459, 0.55151852, 0.63861249]) # 最相似的是第一张范爷,其次是范爷她弟,不愧是亲姐弟啊。
作者:无鱼二饼
链接:https://www.jianshu.com/p/6961c6d9e824