antelopev2-face-features.py

antelopev2-face-features.py

# pip install -U insightface mxnet onnx onnxruntime

# Initialize the FaceAnalysis app with the Antelope model
from insightface.app import FaceAnalysis
app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
# Remove nested folder from download if needed
app.prepare(ctx_id=-1, det_size=(640, 640))

from PIL import Image
img = Image.open('82.png')

# Convert to numpy array
import numpy as np
img = np.array(img)

faces = app.get(img)

# Extract embeddings for each detected face
for face in faces:
    embedding = face.embedding
    embedding.shape # (512,)