Skip to content

Latest commit

 

History

History
39 lines (29 loc) · 1.29 KB

README.md

File metadata and controls

39 lines (29 loc) · 1.29 KB

MobileFaceNet

数据集

本项目提供了标注文件,存放在dataset目录下,解压即可。另外的图片文件需要自行下载,下载解压到dataset目录下即可,如果使用其他数据集,请确保同一个文件夹中的图片是同一个人的,并且是经过对齐裁剪的,大小为112*112

执行下面命令,将提取人脸图片到dataset/images

python create_dataset.py

训练

执行train.py即可,更多训练参数请查看代码。

python train.py

评估

执行eval.py即可,更多训练参数请查看代码。

python eval.py

预测

在执行预测之前,先要在face_db目录下存放人脸图片,每张图片只包含一个人脸,并以该人脸的名称命名,这建立一个人脸库。之后的识别都会跟这些图片对比,找出匹配成功的人脸。

如果是通过图片路径预测的,请执行下面命令。

python infer.py --image_path=temp/test.jpg

如果是通过相机预测的,请执行下面命令。

python infer_camera.py --camera_id=0