Face Identification inference code using ONNX Runtime
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
- Run
python main.py a.jpg b.jpg
for testing
- This repo supports inference only, see reference for more details