| 模型 | 模型下载链接 | 输出特征维度 | AP (%) AgeDB-30/CFP-FP/LFW |
GPU推理耗时 (ms) | CPU推理耗时 | 模型存储大小 (M) | 介绍 |
|---|---|---|---|---|---|---|---|
| MobileFaceNet | 推理模型/训练模型 | 128 | 96.28/96.71/99.58 | 4.1 | 基于MobileFaceNet在MS1Mv3数据集上训练的人脸特征提取模型 | ||
| ResNet50_face | 推理模型/训练模型 | 512 | 98.12/98.56/99.77 | 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。
校验结果文件具体内容为:
{
"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;