当前位置：首页 > news >正文

Python中大名鼎鼎的face_recognition使用

news 来源：原创 2024/5/4 21:04:45

为什么80%的码农都做不了架构师？>>>

背景

face_recognition是号称世界上最简单的人脸识别工具和Python库。虽然，是国外开源的项目（良心的MIT开源协议），竟然有官方的中文文档支持，从未见过如此亲近天朝人民的开源项目了。

问题

手里有一张目标人物图，需要从一堆图片文件中，找出这个人是谁？这一堆的图片文件时，按人的姓名分类的。本来想直接使用face_recognition命令行解决，如下：

face_recognition ./pictures_of_people_i_know/ ./unknown_pictures/

然而，我的unknown_pictures文件下面是按人姓名做为文件夹划分的，face_recognition不支持递归文件夹来找文件，所以，就只能编写python程序来解决了。

Python3

from __future__ import division
from tqdm import tqdm

import os
import face_recognition
import imghdr
import sys


def get_all_files(path_dir):
    all_file = []
    for dir_path, dir_names, filenames in os.walk(path_dir):
        for dir_ in dir_names:
            all_file.append(os.path.join(dir_path, dir_))
        for name in filenames:
            all_file.append(os.path.join(dir_path, name))
    return all_file


def get_image_file(file_check):
    if os.path.isfile(file_check):
        if imghdr.what(file_check) == 'jpeg':
            unknown_picture = face_recognition.load_image_file(file_check)
            unknown_face_encoding_list = face_recognition.face_encodings(unknown_picture)
            if len(unknown_face_encoding_list) > 0:
                unknown_face_encoding = unknown_face_encoding_list[0]
                results = face_recognition.compare_faces([my_face_encoding], unknown_face_encoding, tolerance=0.2)
                if results[0]:
                    return file_check


if __name__ == '__main__':
    picture_of_me = face_recognition.load_image_file(
        "/Users/zhangyalin/Downloads/images/pictures_of_people_i_know/zyl.jpg")
    my_face_encoding = face_recognition.face_encodings(picture_of_me)[0]

    resultList = []

    path = "/Users/zhangyalin/Downloads/images/unknown_pictures/"
    files = get_all_files(path)
    num_tasks = len(files)
    for file in tqdm(files):
        fileTarget = get_image_file(file)
        if fileTarget is not None:
            resultList.append(file)
            break
    print("定位:")
    print(resultList)
    sys.exit()

这里就是递归unknown_pictures文件夹，然后不断将获得的图像中人脸与pictures_of_people_i_know/zyl.jpg图像中人脸进行对比，而zyl.jpg中人叫zyl。具体可以去看Face Recognition 人脸识别，说的很清楚的。

Note:tolerance=0.2，容错率需要设置为0.2，这样人脸识别更加准确。熟悉Python线程池的朋友，可以补充完善一下上面的代码。