--- comments: true --- # 人脸特征模块使用教程 ## 一、概述 人脸特征模块通常以经过检测提取和关键点矫正处理的标准化人脸图像作为输入,从这些图像中提取具有高度辨识性的人脸特征,以便供后续模块使用,如人脸匹配和验证等任务。 ## 二、支持模型列表
| 模型 | 模型下载链接 | 输出特征维度 | Acc (%) AgeDB-30/CFP-FP/LFW |
GPU推理耗时 (ms) | CPU推理耗时 (ms) | 模型存储大小 (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。
## 三、快速集成 > ❗ 在快速集成前,请先安装 PaddleX 的 wheel 包,详细请参考 [PaddleX本地安装教程](../../../installation/installation.md) 完成whl包的安装后,几行代码即可完成人脸特征模块的推理,可以任意切换该模块下的模型,您也可以将人脸特征模块中的模型推理集成到您的项目中。运行以下代码前,请您下载[示例图片](https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/face_recognition_001.jpg)到本地。 ```python from paddlex import create_model model_name = "MobileFaceNet" model = create_model(model_name) output = model.predict("face_recognition_001.jpg", batch_size=1) for res in output: res.print(json_format=False) res.save_to_json("./output/res.json") ```| 参数 | 参数说明 | 参数类型 | 可选项 | 默认值 |
|---|---|---|---|---|
model_name |
模型名称 | str |
无 | 无 |
model_dir |
模型存储路径 | str |
无 | 无 |
flip |
是否进行反转推理; 如果为True,模型会对输入图像水平翻转后再次推理,并融合两次推理结果以提升人脸特征的准确性 | bool |
无 | False |
| 参数 | 参数说明 | 参数类型 | 可选项 | 默认值 |
|---|---|---|---|---|
input |
待预测数据,支持多种输入类型 | Python Var/str/dict/list |
|
无 |
batch_size |
批大小 | int |
任意整数 | 1 |
| 方法 | 方法说明 | 参数 | 参数类型 | 参数说明 | 默认值 |
|---|---|---|---|---|---|
print() |
打印结果到终端 | format_json |
bool |
是否对输出内容进行使用 JSON 缩进格式化 |
True |
indent |
int |
指定缩进级别,以美化输出的 JSON 数据,使其更具可读性,仅当 format_json 为 True 时有效 |
4 | ||
ensure_ascii |
bool |
控制是否将非 ASCII 字符转义为 Unicode。设置为 True 时,所有非 ASCII 字符将被转义;False 则保留原始字符,仅当format_json为True时有效 |
False |
||
save_to_json() |
将结果保存为json格式的文件 | save_path |
str |
保存的文件路径,当为目录时,保存文件命名与输入文件类型命名一致 | 无 |
indent |
int |
指定缩进级别,以美化输出的 JSON 数据,使其更具可读性,仅当 format_json 为 True 时有效 |
4 | ||
ensure_ascii |
bool |
控制是否将非 ASCII 字符转义为 Unicode。设置为 True 时,所有非 ASCII 字符将被转义;False 则保留原始字符,仅当format_json为True时有效 |
False |
| 属性 | 属性说明 |
|---|---|
json |
获取预测的json格式的结果 |
校验结果文件具体内容为:
{
"done_flag": true,
"check_pass": true,
"attributes": {
"train_label_file": "../../dataset/face_rec_examples/train/label.txt",
"train_num_classes": 995,
"train_samples": 1000,
"train_sample_paths": [
"check_dataset/demo_img/01378592.jpg",
"check_dataset/demo_img/04331410.jpg",
"check_dataset/demo_img/03485713.jpg",
"check_dataset/demo_img/02382123.jpg",
"check_dataset/demo_img/01722397.jpg",
"check_dataset/demo_img/02682349.jpg",
"check_dataset/demo_img/00272794.jpg",
"check_dataset/demo_img/03151987.jpg",
"check_dataset/demo_img/01725764.jpg",
"check_dataset/demo_img/02580369.jpg"
],
"val_label_file": "../../dataset/face_rec_examples/val/pair_label.txt",
"val_num_classes": 2,
"val_samples": 500,
"val_sample_paths": [
"check_dataset/demo_img/Don_Carcieri_0001.jpg",
"check_dataset/demo_img/Eric_Fehr_0001.jpg",
"check_dataset/demo_img/Harry_Kalas_0001.jpg",
"check_dataset/demo_img/Francis_Ford_Coppola_0001.jpg",
"check_dataset/demo_img/Amer_al-Saadi_0001.jpg",
"check_dataset/demo_img/Sergei_Ivanov_0001.jpg",
"check_dataset/demo_img/Erin_Runnion_0003.jpg",
"check_dataset/demo_img/Bill_Stapleton_0001.jpg",
"check_dataset/demo_img/Daniel_Bruehl_0001.jpg",
"check_dataset/demo_img/Clare_Short_0004.jpg"
]
},
"analysis": {},
"dataset_path": "./dataset/face_rec_examples",
"show_type": "image",
"dataset_type": "ClsDataset"
}
上述校验结果中,check_pass 为 True 表示数据集格式符合要求,其他部分指标的说明如下:
attributes.train_num_classes:该数据集训练类别数为 995;attributes.val_num_classes:该数据集验证类别数为 2;attributes.train_samples:该数据集训练集样本数量为 1000;attributes.val_samples:该数据集验证集样本数量为 500;attributes.train_sample_paths:该数据集训练集样本可视化图片相对路径列表;人脸特征模块不支持数据格式转换与数据集划分。
output,如需指定保存路径,可通过配置文件中 -o Global.output 字段进行设置。训练其他模型时,需要的指定相应的配置文件,模型和配置的文件的对应关系,可以查阅PaddleX模型列表(CPU/GPU)。
在完成模型训练后,所有产出保存在指定的输出目录(默认为./output/)下,通常有以下产出:
train_result.json:训练结果记录文件,记录了训练任务是否正常完成,以及产出的权重指标、相关文件路径等;
train.log:训练日志文件,记录了训练过程中的模型指标变化、loss 变化等;config.yaml:训练配置文件,记录了本次训练的超参数的配置;.pdparams、.pdema、.pdopt.pdstate、.pdiparams、.pdmodel:模型权重相关文件,包括网络参数、优化器、EMA、静态图网络参数、静态图网络结构等;在模型评估时,需要指定模型权重文件路径,每个配置文件中都内置了默认的权重保存路径,如需要改变,只需要通过追加命令行参数的形式进行设置即可,如-o Evaluate.weight_path=./output/best_model/best_model/model.pdparams。
在完成模型评估后,会产出evaluate_result.json,其记录了评估的结果,具体来说,记录了评估任务是否正常完成,以及模型的评估指标,包含 Accuracy;