/FaceID

Face Identification using ONNX Runtime

Primary LanguagePython

Face Identification inference code using ONNX Runtime

AdaFace: Quality Adaptive Margin for Face Recognition Vizualization

Installation

conda create -n ONNX python=3.8
conda activate ONNX
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
pip install onnxruntime-gpu==1.14.0
pip install opencv-python==4.5.5.64

WebCam Inference

$ python main.py

Model ONNX WEIGHT

Backbone Weight
R50 link

Note

  • This repo supports only inference, see reference for more details

Reference