# 人脸库生成工具 - 使用指南 ## 1. 准备照片 创建目录结构: ``` 数据集目录/ 张三/ 001.jpg 002.jpg 李四/ a.png b.png ``` - 每人一个文件夹,文件夹名就是人名 - 每人建议放 3-10 张照片 - 支持 jpg/png/bmp 格式 ## 2. 生成人脸库 在项目目录打开 PowerShell,运行: ```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"` 为你的图片路径,然后运行: ```powershell testpic.bat ``` 会生成对齐后的图片和特征向量到 `debug_output/` 目录。