Skip to content

Latest commit

 

History

History
9 lines (8 loc) · 697 Bytes

README.md

File metadata and controls

9 lines (8 loc) · 697 Bytes
Raw

face-prediction-rs

Face prediction CLI using rust ORT library, Ultraface and Arcface. Iterates image folder [image_folder] and compares faces found to [test_case_path].

Setup

  1. download the Ultra face 640 onnx model and put it in [ultra_model_path]
  2. download the Arc face arcfaceresnet100-11-int8.onnx model and put it in [arc_model_path]
  3. add images to [image_folder]
  4. run cargo build --release
  5. run ./target/release/face-prediction [ultra_model_path] [arc_model_path] [image_folder] [output_dir] [test_case_path]