VirtuoTuring commited on
Commit
964f474
·
verified ·
1 Parent(s): 4e8308f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +50 -3
README.md CHANGED
@@ -1,3 +1,50 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: onnxruntime
4
+ tags:
5
+ - onnx
6
+ - onnxruntime
7
+ - face-recognition
8
+ - face-embedding
9
+ - computer-vision
10
+ ---
11
+
12
+ # VirtuoTuring Face Embedder
13
+
14
+ ONNX model for generating facial embeddings and comparing two faces.
15
+
16
+ ## What it does
17
+ This model takes an aligned face as input and returns a 512-dimensional facial embedding.
18
+
19
+ ## Expected input
20
+ - RGB image
21
+ - cropped/aligned face
22
+ - size: 112x112
23
+ - type: float32
24
+ - normalization: [-1, 1]
25
+
26
+ ## Output
27
+ - 512-dimensional float32 embedding
28
+
29
+ ## Comparison
30
+ Compare embeddings using cosine similarity.
31
+
32
+ Suggested initial threshold:
33
+ - 0.93
34
+
35
+ ## Limitations
36
+ - it does not perform face detection on its own
37
+ - it assumes the face is already cropped/aligned
38
+ - it is not a certified biometric system
39
+ - it may fail with extreme pose, occlusions, poor lighting, or partial faces
40
+
41
+ ## Files
42
+ - `virtuoturing.onnx`
43
+ - external weights file `.data`
44
+
45
+ ## Usage example
46
+ ```python
47
+ import onnxruntime as ort
48
+ sess = ort.InferenceSession("virtuoturing.onnx")
49
+ print(sess.get_inputs()[0].name, sess.get_inputs()[0].shape)
50
+ print(sess.get_outputs()[0].name, sess.get_outputs()[0].shape)