Back to Insightface

InsightFace Model Zoo

model_zoo/README.md

0.714.4 KB
Original Source

InsightFace Model Zoo

:bell: ALL models are available for non-commercial research purposes only.

0. Python Package models

To check the detail of insightface python package, please see here.

To install: pip install -U insightface

To use the specific model pack:

model_pack_name = 'buffalo_l'
app = FaceAnalysis(name=model_pack_name)

Name in bold is the default model pack in latest version.

NameDetection ModelRecognition ModelAlignmentAttributesModel-Size
antelopev2RetinaFace-10GFResNet100@Glint360K2d106 & 3d68Gender&Age407MB
buffalo_lRetinaFace-10GFResNet50@WebFace600K2d106 & 3d68Gender&Age326MB
buffalo_mRetinaFace-2.5GFResNet50@WebFace600K2d106 & 3d68Gender&Age313MB
buffalo_sRetinaFace-500MFMBF@WebFace600K2d106 & 3d68Gender&Age159MB
buffalo_scRetinaFace-500MFMBF@WebFace600K--16MB

Recognition accuracy of python library model packs:

NameMR-ALLAfricanCaucasianSouth AsianEast AsianLFWCFP-FPAgeDB-30IJB-C(E4)
buffalo_l91.2590.2994.7093.1674.9699.8399.3398.2397.25
buffalo_s71.8769.4580.4573.3951.0399.7098.0096.5895.02

buffalo_m has the same accuracy with buffalo_l.

buffalo_sc has the same accuracy with buffalo_s.

(Note that almost all ONNX models in our model_zoo can be called by python library.)

1. Face Recognition models.

Definition:

The default training loss is margin based softmax if not specified.

MFN: MobileFaceNet

MS1MV2: MS1M-ArcFace

MS1MV3: MS1M-RetinaFace

MS1M_MegaFace: MS1MV2+MegaFace_train

_pfc: using Partial FC, with sample-ratio=0.1

MegaFace: MegaFace identification test, with gallery=1e6.

IJBC: IJBC 1:1 test, under FAR<=1e-4.

BDrive: BaiduDrive

GDrive: GoogleDrive

List of models by MXNet and PaddlePaddle:

BackboneDatasetMethodLFWCFP-FPAgeDB-30MegaFaceLink.
R100 (mxnet)MS1MV2ArcFace99.7798.2798.2898.47BDrive, GDrive
MFN (mxnet)MS1MV1ArcFace99.5088.9495.91-BDrive, GDrive
MFN (paddle)MS1MV2ArcFace99.4593.4396.13-pretrained model, inference model
iResNet50 (paddle)MS1MV2ArcFace99.7397.4397.88-pretrained model, inference model

List of models by various depth IResNet and training datasets:

BackboneDatasetMR-ALLAfricanCaucasianSouth AsianEast AsianLink(onnx)
R100Casia42.73539.66653.93347.80721.572GDrive
R100MS1MV280.72579.11787.17685.50155.807GDrive
R18MS1MV368.32662.61375.12570.21343.859GDrive
R34MS1MV377.36571.64483.29180.08453.712GDrive
R50MS1MV380.53375.48886.11584.30557.352GDrive
R100MS1MV384.31281.08389.04088.08262.193GDrive
R18Glint360K72.07468.23080.57575.85247.831GDrive
R34Glint360K83.01579.90788.62086.81560.604GDrive
R50Glint360K87.07785.27291.61790.54166.813GDrive
R100Glint360K90.65989.48894.28593.43472.528GDrive

List of models by IResNet-50 and different training datasets:

DatasetMR-ALLAfricanCaucasianSouth AsianEast AsianLFWCFP-FPAgeDB-30IJB-C(E4)Link(onnx)
CISIA36.79442.55055.82549.61819.61199.45095.21494.90087.220GDrive
CISIA_pfc37.10738.93453.82348.67419.92799.36795.42994.60084.970GDrive
VGG238.57835.25954.30444.08124.09599.55097.41095.08091.220GDrive
VGG2_pfc40.67336.76760.18049.03924.25599.68398.52995.40092.490GDrive
GlintAsia62.66349.53164.82957.98461.74399.58393.18695.40091.500GDrive
GlintAsia_pfc63.14950.36665.22757.93661.82099.65093.02995.23391.140GDrive
MS1MV277.69674.59684.12682.04151.10599.83398.08398.08396.140GDrive
MS1MV2_pfc77.73874.72884.88382.79852.50799.78398.07198.01796.080GDrive
MS1M_MegaFace78.37274.13882.25177.22360.20399.75097.55797.40095.350GDrive
MS1M_MegaFace_pfc78.77373.69082.94778.79357.56699.80097.87097.73395.400GDrive
MS1MV382.52277.17287.02886.00660.62599.80098.52998.26796.580GDrive
MS1MV3_pfc81.68378.12687.28685.54258.92599.80098.44398.16796.430GDrive
Glint360k86.78984.74991.41490.08866.16899.81799.14398.45097.130GDrive
Glint360k_pfc87.07785.27291.61690.54166.81399.81799.14398.45097.020GDrive
WebFace600K90.56689.35594.17792.35873.85299.80099.20098.10097.120GDrive
WebFace600K_pfc89.95189.30194.01692.38173.00799.81799.14398.11797.010GDrive
Average69.24765.90877.12172.81952.01499.70697.37496.96293.925
Average_pfc69.51965.89877.49773.21351.85399.71597.45796.96593.818

List of models by MobileFaceNet and different training datasets:

FLOPS: 450M FLOPs

Model-Size: 13MB

DatasetMR-ALLAfricanCaucasianSouth AsianEast AsianLFWCFP-FPAgeDB-30IJB-C(E4)Link(onnx)
WebFace600K71.86569.44980.45473.39451.02699.7098.0096.5895.02-

2. Face Detection models.

2.1 RetinaFace

In RetinaFace, mAP was evaluated with multi-scale testing.

m025: means MobileNet-0.25

ImpelmentationEasy-SetMedium-SetHard-SetLink
RetinaFace-R5096.595.690.4BDrive, GDrive
RetinaFace-m025(yangfly)--82.5BDrive(nzof), GDrive
BlazeFace-FPN-SSH (paddle)91.989.881.7%pretrained model, inference model

2.2 SCRFD

In SCRFD, mAP was evaluated with single scale testing, VGA resolution.

2.5G: means the model cost 2.5G FLOPs while the input image is in VGA(640x480) resolution.

_KPS: means this model can detect five facial keypoints.

NameEasyMediumHardFLOPsParams(M)Infer(ms)Link(pth)
SCRFD_500M90.5788.1268.51500M0.573.6GDrive
SCRFD_1G92.3890.5774.801G0.644.1GDrive
SCRFD_2.5G93.7892.1677.872.5G0.674.2GDrive
SCRFD_10G95.1693.8783.0510G3.864.9GDrive
SCRFD_34G96.0694.9285.2934G9.8011.7GDrive
SCRFD_500M_KPS90.9788.4469.49500M0.573.6GDrive
SCRFD_2.5G_KPS93.8092.0277.132.5G0.824.3GDrive
SCRFD_10G_KPS95.4094.0182.8010G4.235.0GDrive

3. Face Alignment models.

2.1 2D Face Alignment

ImpelmentationPointsBackboneParams(M)Link(onnx)
Coordinate-regression106MobileNet-0.51.2GDrive

2.2 3D Face Alignment

ImpelmentationPointsBackboneParams(M)Link(onnx)
-68ResNet-5034.2GDrive

2.3 Dense Face Alignment

4. Face Attribute models.

4.1 Gender&Age

Training-SetBackboneParams(M)Link(onnx)
CelebAMobileNet-0.250.3GDrive

4.2 Expression