MobileFaceNet_TF
Tensorflow implementation for MobileFaceNet.
dependencies
- tensorflow >= r1.5
- opencv-python 3.x
- python 3.x
- scipy
- sklearn
- numpy
- mxnet
- pickle
Prepare dataset
- choose one of the following links to download dataset which is provide by insightface. (Special Recommend MS1M-refine-v2)
- MS1M-refine-v2@BaiduDrive, MS1M-refine-v2@GoogleDrive
- Refined-MS1M@BaiduDrive, Refined-MS1M@GoogleDrive
- VGGFace2@BaiduDrive, VGGFace2@GoogleDrive
- Insightface Dataset Zoo
- move dataset to
${MobileFaceNet_TF_ROOT}/datasets
. - run
${MobileFaceNet_TF_ROOT}/utils/data_process.py
.
pretrained model
training
- refined super parameters by yourself special project.
- run script
${MobileFaceNet_TF_ROOT}/train_nets.py
- have a snapshot result at
${MobileFaceNet_TF_ROOT}/output
.
performance
size | LFW(%) | Val@1e-3(%) | inference@MSM8976-cpu(ms) |
---|---|---|---|
5.7M | 99.4+ | 98.4+ | 260- |