Add ArcFace face recognition model

Kotlin / kotlindl

High-level Deep Learning Framework written in Kotlin and inspired by Keras

Apache License 2.0

1.44k stars 103 forks source link

Add ArcFace face recognition model #495

Open StefanOltmann opened 1 year ago

StefanOltmann commented 1 year ago

It's great that we have support for UltraFace to detect the faces in an image!

I want to go one step further and match the faces to persons, so my app can suggest person tags for images or even create automatic person photo albums.

As far as I understand I will need ArcFace for that.

The ONNX model is located here: https://github.com/onnx/models/tree/main/validated/vision/body_analysis/arcface

Would be great to have it supported in a similar way like UltraFace.

StefanOltmann commented 6 months ago

@zaleslaw I come to the point where I need this for my app.

What’s the current state regarding this?