Pytorch0.4.1 codes for InsightFace
Alternatives To Insightface_pytorch
Project NameStarsDownloadsRepos Using ThisPackages Using ThisMost Recent CommitTotal ReleasesLatest ReleaseOpen IssuesLicenseLanguage
Pytorch Cyclegan And Pix2pix20,036
4 months ago493otherPython
Image-to-Image Translation in PyTorch
Insightface18,28619a day ago28December 17, 2022982mitPython
State-of-the-art 2D and 3D Face Analysis Project
Datasets17,218954020 hours ago69July 31, 2023596apache-2.0Python
🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools
Vision14,5872,3062,088a day ago37May 08, 2023941bsd-3-clausePython
Datasets, Transforms and Models specific to Computer Vision
Cvat10,16023 days ago11July 28, 2023546mitTypeScript
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
Deeplake6,91111a day ago93August 05, 202362mpl-2.0Python
Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stream data in real-time to PyTorch/TensorFlow.
11 days ago1apache-2.0
Techniques for deep learning with satellite & aerial imagery
9 months ago7mpl-2.0Jupyter Notebook
:robot: :speech_balloon: Deep learning for Text to Speech (Discussion forum:
Transformers Tutorials5,688
3 days ago210mitJupyter Notebook
This repository contains demos I made with the Transformers library by HuggingFace.
Yet Another Efficientdet Pytorch4,892
2 years ago325lgpl-3.0Jupyter Notebook
The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
Alternatives To Insightface_pytorch
Select To Compare

Alternative Project Comparisons


Pytorch0.4.1 codes for InsightFace

1. Intro

  • This repo is a reimplementation of Arcface(paper), or Insightface(github)
  • For models, including the pytorch implementation of the backbone modules of Arcface and MobileFacenet
  • Codes for transform MXNET data records in Insightface(github) to Image Datafolders are provided
  • Pretrained models are posted, include the MobileFacenet and IR-SE50 in the original paper

2. Pretrained Models & Performance

IR-SE50 @ BaiduNetdisk, IR-SE50 @ Onedrive

LFW(%) CFP-FF(%) CFP-FP(%) AgeDB-30(%) calfw(%) cplfw(%) vgg2_fp(%)
0.9952 0.9962 0.9504 0.9622 0.9557 0.9107 0.9386

Mobilefacenet @ BaiduNetDisk, Mobilefacenet @ OneDrive

LFW(%) CFP-FF(%) CFP-FP(%) AgeDB-30(%) calfw(%) cplfw(%) vgg2_fp(%)
0.9918 0.9891 0.8986 0.9347 0.9402 0.866 0.9100

3. How to use

  • clone

    git clone

3.1 Data Preparation

3.1.1 Prepare Facebank (For testing over camera or video)

Provide the face images your want to detect in the data/face_bank folder, and guarantee it have a structure like following:

        ---> id1/
            ---> id1_1.jpg
        ---> id2/
            ---> id2_1.jpg
        ---> id3/
            ---> id3_1.jpg
           ---> id3_2.jpg

3.1.2 download the pretrained model to work_space/model

If more than 1 image appears in one folder, an average embedding will be calculated

3.2.3 Prepare Dataset ( For training)

download the refined dataset: (emore recommended)

Note: If you use the refined MS1M dataset and the cropped VGG2 dataset, please cite the original papers.

  • after unzip the files to 'data' path, run :


    after the execution, you should find following structure:

            ---> agedb_30
            ---> calfw
            ---> cfp_ff
            --->  cfp_fp
            ---> cfp_fp
            ---> cplfw
            ---> lfw
            ---> vgg2_fp

3.2 detect over camera:

- facebank/
    if more than 1 image appears in the directory, average embedding will be calculated
  • 4 to start


3.3 detect over video:

python -f [video file name] -s [save file name]

the video file should be inside the data/face_bank folder

3.4 Training:

python -b [batch_size] -lr [learning rate] -e [epochs]

# python -net mobilefacenet -b 200 -w 4

4. References


  • PRs are welcome, in case that I don't have the resource to train some large models like the 100 and 151 layers model
  • Email : [email protected]
Popular Dataset Projects
Popular Pytorch Projects
Popular Data Processing Categories

Get A Weekly Email With Trending Projects For These Categories
No Spam. Unsubscribe easily at any time.
Jupyter Notebook