insightface 提取人臉特征（超簡潔）

insightface 是一款優秀的開源人臉檢測/識別庫，下面將展示如何使用它

安裝：

pip install insightface -i https://mirror.baidu.com/pypi/simple

 

如需使用GPU資源（可選）：

pip install onnxruntime-gpu==1.3.0 -i https://mirror.baidu.com/pypi/simple

我的cuda是10.1的，請選擇對應版本安裝

示例：

import insightface
import cv2


model = insightface.app.FaceAnalysis()
model.prepare(ctx_id=0, det_thresh=0.45)

face_img = cv2.imread('test.jpg')
rgb_small_frame = face_img[:, :, ::-1]
res = model.get(rgb_small_frame)

print('人臉數量：', len(res))
print('res: ', res[0].keys())  # 結果包括 3d-points, 2d-points, age, gender, box, feature