insightface使用笔记

InsightFace 高效地实现了各种人脸识别、人脸检测和人脸对齐算法,这些算法针对训练和部署进行了优化。

 insightface 使用 onnxruntime 作为推理后端。 手动安装 onnxruntime-gpu 使用 GPU ,或者安装 onnxruntime 使用cpu

样例

import cv2
import numpy as np
import insightface
from insightface.app import FaceAnalysis
from insightface.data import get_image as ins_get_image

app = FaceAnalysis(providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
app.prepare(ctx_id=0, det_size=(640, 640))
img = ins_get_image('t1')
faces = app.get(img)
rimg = app.draw_on(img, faces)
cv2.imwrite("./t1_output.jpg", rimg)

代码将从 t1.jpg 图像中检测人脸,并在其上绘制检测结果。

模型

默认模型是buffalo_l, auto列表示是否自动下载,手动下载的,放到 ~/.insightface/models/ 下

NameDetection ModelRecognition ModelAlignmentAttributesModel-SizeLinkAuto
antelopev2SCRFD-10GFResNet100@Glint360K2d106 & 3d68Gender&Age407MBlinkN
buffalo_lSCRFD-10GFResNet50@WebFace600K2d106 & 3d68Gender&Age326MBlinkY
buffalo_mSCRFD-2.5GFResNet50@WebFace600K2d106 & 3d68Gender&Age313MBlinkN
buffalo_sSCRFD-500MFMBF@WebFace600K2d106 & 3d68Gender&Age159MBlinkN
buffalo_scSCRFD-500MFMBF@WebFace600K--16MBlinkN

识别准确率

NameMR-ALL非洲白人南亚东亚LFWCFP-FPAgeDB-30IJB-C(E4)
buffalo_l91.2590.2994.7093.1674.9699.8399.3398.2397.25
buffalo_s71.8769.4580.4573.3951.0399.7098.0096.5895.02

调用检测模型

import cv2
import numpy as np
import insightface
from insightface.app import FaceAnalysis
from insightface.data import get_image as ins_get_image

# Method-1, use FaceAnalysis
app = FaceAnalysis(allowed_modules=['detection']) # enable detection model only
app.prepare(ctx_id=0, det_size=(640, 640))

# Method-2, load model directly
detector = insightface.model_zoo.get_model('your_detection_model.onnx')
detector.prepare(ctx_id=0, input_size=(640, 640))

调用识别模型

import cv2
import numpy as np
import insightface
from insightface.app import FaceAnalysis
from insightface.data import get_image as ins_get_image

handler = insightface.model_zoo.get_model('your_recognition_model.onnx')
handler.prepare(ctx_id=0)

参考:

GitHub - deepinsight/insightface: State-of-the-art 2D and 3D Face Analysis Project

基于insightface实现的人脸识别和人脸注册_夜雨飘零1的博客-CSDN博客

 https://www.cnblogs.com/divenswu/p/16450243.html

https://github.com/deepinsight/insightface/tree/0cb03d391bb9335821448ce6e0519dbc870ec68d/examples

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值