人脸库生成工具 - 使用指南

1. 准备照片

创建目录结构：

数据集目录/
  张三/
    001.jpg
    002.jpg
  李四/
    a.png
    b.png

每人一个文件夹，文件夹名就是人名
每人建议放 3-10 张照片
支持 jpg/png/bmp 格式

2. 生成人脸库

在项目目录打开 PowerShell，运行：

build_gallery.bat `
  --dataset "D:\你的数据集目录" `
  --db_out "D:\输出目录\face_gallery.db" `
  --det_model "./models/RetinaFace_mobile320.onnx" `
  --recog_model "./models/mobilefacenet_arcface_prenorm.onnx" `
  --det_outputs_config "./models/retinaface_mobile320_config.json"

只需修改 D:\你的数据集目录 和 D:\输出目录 两个路径即可。

3. 查看结果

生成成功后，会输出统计信息：

总人数、成功入库人数
每人使用的照片数
失败原因统计

生成的 face_gallery.db 就是最终的人脸库文件。

可选参数

参数	默认值	说明
`--max_imgs_per_person`	10	每人最多用几张照片
`--min_face_size`	80	人脸最小尺寸（像素）
`--pick_face`	largest	多脸时选哪个：largest(最大)/first(第一个)/highest_score(置信度最高)

测试单张图片

修改 testpic.py 第 9 行的 INPUT_IMAGE = "003.jpg" 为你的图片路径，然后运行：

testpic.bat

会生成对齐后的图片和特征向量到 debug_output/ 目录。

1.5 KiB Raw Permalink Blame History Unescape Escape