olalium / 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].

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]