MitchellX / FaceX-Zoo

A PyTorch Toolbox for Face Recognition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FaceX-Zoo

FaceX-Zoo is a PyTorch toolbox for face recognition. It provides a training module with various supervisory heads and backbones towards state-of-the-art face recognition, as well as a standardized evaluation module which enables to evaluate the models in most of the popular benchmarks just by editing a simple configuration. Also, a simple yet fully functional face SDK is provided for the validation and primary application of the trained models. Rather than including as many as possible of the prior techniques, we enable FaceX-Zoo to easilyupgrade and extend along with the development of face related domains. Please refer to the technical report for more detailed information about this project.

About the name:

  • "Face" - this repo is mainly for face recognition.
  • "X" - we also aim to provide something beyond face recognition, e.g. face paring, face lightning.
  • "Zoo" - there include a lot of algorithms and models in this repo. image

What's New

  • [Jan. 2021] We commit the initial version of FaceX-Zoo.

Requirements

  • python >= 3.7.1
  • pytorch >= 1.1.0
  • torchvision >= 0.3.0

Model Training

See README.md in training_mode, currently support conventional training and semi-siamese training.

Model Evaluation

See README.md in test_protocol, currently support LFW, CPLFW, CALFT, RFW, AgeDB, MegaFace and MegaFace-mask.

Face SDK

See README.md in face_sdk, currently support face detection, face alignment and face recognition.

Face Mask Adding

See README.md in FMA-3D.

License

FaceX-Zoo is released under the Apache License, Version 2.0.

References

This repo is mainly inspired by InsightFace, InsightFace_Pytorch, face.evoLVe.

Citing FaceX-Zoo

If you use FaceX-Zoo in your research or wish to refer to the baseline results published by this project, please use the following BibTeX entry.

@article{wang2021facex-zoo,
  title={FaceX-Zoo: A PyTorch Toolbox for Face Recognition},
  author={Wang, Jun and Liu, Yinglu and Hu, Yibo and Shi, Hailin and Mei, Tao},
  journal={arXix preprint arXiv:2101.04407},
  year={2021}
}

About

A PyTorch Toolbox for Face Recognition

License:Apache License 2.0


Languages

Language:Python 94.7%Language:C++ 4.6%Language:Shell 0.8%