---
comments: true
---
# 人脸识别产线使用教程
## 1. 人脸识别产线介绍
人脸识别任务是计算机视觉领域的重要组成部分,旨在通过分析和比较人脸特征,实现对个人身份的自动识别。该任务不仅需要检测图像中的人脸,还需要对人脸图像进行特征提取和匹配,从而在数据库中找到对应的身份信息。人脸识别广泛应用于安全认证、监控系统、社交媒体和智能设备等场景。
人脸识别产线是专注于解决人脸定位和识别任务的端到端串联系统,可以从图像中快速准确地定位人脸区域、提取人脸特征,并与特征库中预先建立的特征做检索比对,从而确认身份信息。
人脸识别产线中包含了人脸检测模块和人脸特征模块,每个模块中包含了若干模型,具体使用哪些模型,您可以根据下边的 benchmark 数据来选择。如您更考虑模型精度,请选择精度较高的模型,如您更考虑模型推理速度,请选择推理速度较快的模型,如您更考虑模型存储大小,请选择存储大小较小的模型。
人脸检测模块:
| 模型 | 模型下载链接 | AP (%) Easy/Medium/Hard |
GPU推理耗时 (ms) | CPU推理耗时 (ms) | 模型存储大小 (M) | 介绍 |
|---|---|---|---|---|---|---|
| BlazeFace | 推理模型/训练模型 | 77.7/73.4/49.5 | 49.9 | 68.2 | 0.447 | 轻量高效的人脸检测模型 |
| BlazeFace-FPN-SSH | 推理模型/训练模型 | 83.2/80.5/60.5 | 52.4 | 73.2 | 0.606 | BlazeFace的改进模型,增加FPN和SSH结构 |
| PicoDet_LCNet_x2_5_face | 推理模型/训练模型 | 93.7/90.7/68.1 | 33.7 | 185.1 | 28.9 | 基于PicoDet_LCNet_x2_5的人脸检测模型 |
| PP-YOLOE_plus-S_face | 推理模型/训练模型 | 93.9/91.8/79.8 | 25.8 | 159.9 | 26.5 | 基于PP-YOLOE_plus-S的人脸检测模型 |
注:以上精度指标是在WIDER-FACE验证集上,以640 *640作为输入尺寸评估得到的。所有模型 GPU 推理耗时基于 NVIDIA V100 机器,精度类型为 FP32, CPU 推理速度基于 Intel(R) Xeon(R) Gold 6271C CPU @ 2.60GHz,精度类型为 FP32。
人脸特征模块:
| 模型 | 模型下载链接 | 输出特征维度 | Acc (%) AgeDB-30/CFP-FP/LFW |
GPU推理耗时 (ms) | CPU推理耗时 | 模型存储大小 (M) | 介绍 |
|---|---|---|---|---|---|---|---|
| MobileFaceNet | 推理模型/训练模型 | 128 | 96.28/96.71/99.58 | 5.7 | 101.6 | 4.1 | 基于MobileFaceNet在MS1Mv3数据集上训练的人脸特征提取模型 |
| ResNet50_face | 推理模型/训练模型 | 512 | 98.12/98.56/99.77 | 8.7 | 200.7 | 87.2 | 基于ResNet50在MS1Mv3数据集上训练的人脸特征提取模型 |
注:以上精度指标是分别在 AgeDB-30、CFP-FP 和 LFW 数据集上测得的 Accuracy。所有模型 GPU 推理耗时基于 NVIDIA Tesla T4 机器,精度类型为 FP32, CPU 推理速度基于 Intel(R) Xeon(R) Gold 5117 CPU @ 2.00GHz,线程数为8,精度类型为 FP32。
## 2. 快速开始 PaddleX 所提供的预训练的模型产线均可以快速体验效果,你可以在线体验人脸识别产线的效果,也可以在本地使用命令行或 Python 体验人脸识别产线的效果。 ### 2.1 在线体验 暂不支持在线体验 ### 2.2 本地体验 > ❗ 在本地使用人脸识别产线前,请确保您已经按照[PaddleX安装教程](../../../installation/installation.md)完成了PaddleX的wheel包安装。 #### 2.2.1 命令行方式体验 暂不支持命令行体验 默认使用内置的的通用图像识别产线配置文件,若您需要自定义配置文件,可执行如下命令获取:paddlex --get_pipeline_config face_recognition
执行后,通用图像识别产线配置文件将被保存在当前路径。若您希望自定义保存位置,可执行如下命令(假设自定义保存位置为./my_path):
paddlex --get_pipeline_config face_recognition --save_path ./my_path
| 参数 | 参数说明 | 参数类型 | 默认值 |
|---|---|---|---|
pipeline |
产线名称或是产线配置文件路径。如为产线名称,则必须为 PaddleX 所支持的产线。 | str |
无 |
index |
产线推理预测所用的特征库,支持:1. str类型表示的目录(该目录下需要包含索引库文件,包括vector.index和index_info.yaml);2. IndexData对象。如不传入该参数,则需要在predict()中指定index。 |
str |
None |
device |
产线模型推理设备。支持:“gpu”,“cpu”。 | str |
gpu |
use_hpip |
是否启用高性能推理,仅当该产线支持高性能推理时可用。 | bool |
False |
| 参数 | 参数说明 | 参数类型 | 默认值 |
|---|---|---|---|
gallery_imgs |
要添加的底库图片,支持:1. str类型表示的图片根目录,数据组织方式与构建索引库时相同,参考2.3节 构建特征库的数据组织方式;2. [numpy.ndarray, numpy.ndarray, ..]类型的底库图片数据。 |
str|list |
无 |
gallery_label |
底库图片的标注信息,支持:1. str类型表示的标注文件的路径,数据组织方式与构建特征库时相同,参考2.3节 构建特征库的数据组织方式;2. [str, str, ..]类型表示的底库图片标注。 |
str |
无 |
| 参数 | 参数说明 | 参数类型 | 默认值 |
|---|---|---|---|
save_path |
特征库文件的保存目录,如drink_index。 |
str |
无 |
| 参数类型 | 参数说明 |
|---|---|
| Python Var | 支持直接传入Python变量,如numpy.ndarray表示的图像数据。 |
| str | 支持传入待预测数据文件路径,如图像文件的本地路径:/root/data/img.jpg。 |
| str | 支持传入待预测数据文件URL,如图像文件的网络URL:示例。 |
| str | 支持传入本地目录,该目录下需包含待预测数据文件,如本地路径:/root/data/。 |
| dict | 支持传入字典类型,字典的key需与具体任务对应,如图像分类任务对应\"img\",字典的val支持上述类型数据,例如:{\"img\": \"/root/data1\"}。 |
| list | 支持传入列表,列表元素需为上述类型数据,如[numpy.ndarray, numpy.ndarray],[\"/root/data/img1.jpg\", \"/root/data/img2.jpg\"],[\"/root/data1\", \"/root/data2\"],[{\"img\": \"/root/data1\"}, {\"img\": \"/root/data2/img.jpg\"}]。 |
| 参数类型 | 参数说明 |
|---|---|
index |
产线推理预测所用的特征库,支持:1. str类型表示的目录(该目录下需要包含特征库文件,包括vector.index和index_info.yaml);2. IndexData对象。如不传入该参数,则默认使用在create_pipeline()中通过参数index指定的特征库。 |
| 方法 | 说明 | 方法参数 |
|---|---|---|
| 打印结果到终端 | - format_json:bool类型,是否对输出内容进行使用json缩进格式化,默认为True;- indent:int类型,json格式化设置,仅当format_json为True时有效,默认为4;- ensure_ascii:bool类型,json格式化设置,仅当format_json为True时有效,默认为False; |
|
| save_to_json | 将结果保存为json格式的文件 | - save_path:str类型,保存的文件路径,当为目录时,保存文件命名与输入文件类型命名一致;- indent:int类型,json格式化设置,默认为4;- ensure_ascii:bool类型,json格式化设置,默认为False; |
| save_to_img | 将结果保存为图像格式的文件 | - save_path:str类型,保存的文件路径,当为目录时,保存文件命名与输入文件类型命名一致; |
| 参数 | 参数说明 | 参数类型 | 默认值 |
|---|---|---|---|
gallery_imgs |
要添加的底库图片,支持:1. str类型表示的图片根目录,数据组织方式与构建索引库时相同,参考2.3节 构建特征库的数据组织方式;2. [numpy.ndarray, numpy.ndarray, ..]类型的底库图片数据。 |
str|list |
无 |
gallery_label |
底库图片的标注信息,支持:1. str类型表示的标注文件的路径,数据组织方式与构建特征库时相同,参考2.3节 构建特征库的数据组织方式;2. [str, str, ..]类型表示的底库图片标注。 |
str|list |
无 |
remove_ids |
待删除的索引序号,支持:1. str类型表示的txt文件的路径,内容为待删除的索引id,每行一个“id”;2. [int, int, ..]类型表示的待删除的索引序号。仅在 remove_index 中有效。 |
str|list |
无 |
index |
特征库,支持:1. 特征库文件(vector.index和index_info.yaml)所在目录的路径;2. IndexData类型的特征库对象,仅在 append_index 和 remove_index 中有效,表示待修改的特征库。 |
str|IndexData |
无 |
index_type |
支持 HNSW32、IVF、Flat。其中,HNSW32 检索速度较快且精度较高,但不支持 remove_index() 操作;IVF 检索速度较快但精度相对较低,支持 append_index() 和 remove_index() 操作;Flat 检索速度较低精度较高,支持 append_index() 和 remove_index() 操作。 |
str |
HNSW32 |
metric_type |
支持:IP,内积(Inner Product);L2,欧几里得距离(Euclidean Distance)。 |
str |
IP |
服务提供的主要操作如下:
buildIndex构建特征向量索引。
POST /face-recognition-index-build
| 名称 | 类型 | 含义 | 是否必填 |
|---|---|---|---|
imageLabelPairs |
array |
用于构建索引的图像-标签对。 | 是 |
imageLabelPairs中的每个元素为一个object,具有如下属性:
| 名称 | 类型 | 含义 |
|---|---|---|
image |
string |
服务可访问的图像文件的URL或图像文件内容的Base64编码结果。 |
label |
string |
标签。 |
result具有如下属性:| 名称 | 类型 | 含义 |
|---|---|---|
indexKey |
string |
索引对应的键,用于标识建立的索引。可用作其他操作的输入。 |
idMap |
object |
向量ID到标签的映射。 |
addImagesToIndex将图像(对应的特征向量)加入索引。
POST /face-recognition-index-add
| 名称 | 类型 | 含义 | 是否必填 |
|---|---|---|---|
imageLabelPairs |
array |
用于构建索引的图像-标签对。 | 是 |
indexKey |
string |
索引对应的键。由buildIndex操作提供。 |
是 |
imageLabelPairs中的每个元素为一个object,具有如下属性:
| 名称 | 类型 | 含义 |
|---|---|---|
image |
string |
服务可访问的图像文件的URL或图像文件内容的Base64编码结果。 |
label |
string |
标签。 |
result具有如下属性:| 名称 | 类型 | 含义 |
|---|---|---|
idMap |
object |
向量ID到标签的映射。 |
removeImagesFromIndex从索引中移除图像(对应的特征向量)。
POST /face-recognition-index-remove
| 名称 | 类型 | 含义 | 是否必填 |
|---|---|---|---|
ids |
array |
需要从索引中移除的向量的ID。 | 是 |
indexKey |
string |
索引对应的键。由buildIndex操作提供。 |
是 |
result具有如下属性:| 名称 | 类型 | 含义 |
|---|---|---|
idMap |
object |
向量ID到标签的映射。 |
infer进行图像识别。
POST /face-recognition-infer
| 名称 | 类型 | 含义 | 是否必填 |
|---|---|---|---|
image |
string |
服务可访问的图像文件的URL或图像文件内容的Base64编码结果。 | 是 |
indexKey |
string |
索引对应的键。由buildIndex操作提供。 |
否 |
result具有如下属性:| 名称 | 类型 | 含义 |
|---|---|---|
faces |
array |
检测到的人脸的信息。 |
image |
string |
识别结果图。图像为JPEG格式,使用Base64编码。 |
faces中的每个元素为一个object,具有如下属性:
| 名称 | 类型 | 含义 |
|---|---|---|
bbox |
array |
人脸目标位置。数组中元素依次为边界框左上角x坐标、左上角y坐标、右下角x坐标以及右下角y坐标。 |
recResults |
array |
识别结果。 |
score |
number |
检测得分。 |
recResults中的每个元素为一个object,具有如下属性:
| 名称 | 类型 | 含义 |
|---|---|---|
label |
string |
标签。 |
score |
number |
识别得分。 |
import base64
import pprint
import sys
import requests
API_BASE_URL = "http://0.0.0.0:8080"
base_image_label_pairs = [
{"image": "./demo0.jpg", "label": "ID0"},
{"image": "./demo1.jpg", "label": "ID1"},
{"image": "./demo2.jpg", "label": "ID2"},
]
image_label_pairs_to_add = [
{"image": "./demo3.jpg", "label": "ID2"},
]
ids_to_remove = [1]
infer_image_path = "./demo4.jpg"
output_image_path = "./out.jpg"
for pair in base_image_label_pairs:
with open(pair["image"], "rb") as file:
image_bytes = file.read()
image_data = base64.b64encode(image_bytes).decode("ascii")
pair["image"] = image_data
payload = {"imageLabelPairs": base_image_label_pairs}
resp_index_build = requests.post(f"{API_BASE_URL}/face-recognition-index-build", json=payload)
if resp_index_build.status_code != 200:
print(f"Request to face-recognition-index-build failed with status code {resp_index_build}.")
pprint.pp(resp_index_build.json())
sys.exit(1)
result_index_build = resp_index_build.json()["result"]
print(f"Number of images indexed: {len(result_index_build['idMap'])}")
for pair in image_label_pairs_to_add:
with open(pair["image"], "rb") as file:
image_bytes = file.read()
image_data = base64.b64encode(image_bytes).decode("ascii")
pair["image"] = image_data
payload = {"imageLabelPairs": image_label_pairs_to_add, "indexKey": result_index_build["indexKey"]}
resp_index_add = requests.post(f"{API_BASE_URL}/face-recognition-index-add", json=payload)
if resp_index_add.status_code != 200:
print(f"Request to face-recognition-index-add failed with status code {resp_index_add}.")
pprint.pp(resp_index_add.json())
sys.exit(1)
result_index_add = resp_index_add.json()["result"]
print(f"Number of images indexed: {len(result_index_add['idMap'])}")
payload = {"ids": ids_to_remove, "indexKey": result_index_build["indexKey"]}
resp_index_remove = requests.post(f"{API_BASE_URL}/face-recognition-index-remove", json=payload)
if resp_index_remove.status_code != 200:
print(f"Request to face-recognition-index-remove failed with status code {resp_index_remove}.")
pprint.pp(resp_index_remove.json())
sys.exit(1)
result_index_remove = resp_index_remove.json()["result"]
print(f"Number of images indexed: {len(result_index_remove['idMap'])}")
with open(infer_image_path, "rb") as file:
image_bytes = file.read()
image_data = base64.b64encode(image_bytes).decode("ascii")
payload = {"image": image_data, "indexKey": result_index_build["indexKey"]}
resp_infer = requests.post(f"{API_BASE_URL}/face-recognition-infer", json=payload)
if resp_infer.status_code != 200:
print(f"Request to face-recogntion-infer failed with status code {resp_infer}.")
pprint.pp(resp_infer.json())
sys.exit(1)
result_infer = resp_infer.json()["result"]
with open(output_image_path, "wb") as file:
file.write(base64.b64decode(result_infer["image"]))
print(f"Output image saved at {output_image_path}")
print("\nDetected faces:")
pprint.pp(result_infer["faces"])