insightface提取embedding"/>
insightface提取embedding
最近在复现一篇文章(Mate Portrait)的训练模型,需要自己提取人脸的特征向量,按照作者给的参考,用insightface提取。
1. 安装
安装MXNex:insightface时基于MXNet框架开发的
pip install mxnet-cu101 # 如果有GPU
pip install mxnet # 如果没有GPU
安装opencv
pip install opencv-contrib-python
安装onnxruntime
pip install onnxruntime-gpu
安装insightface
pip install insightface
2. 提取embedding
一个坑:cv2不能读png,所以不要用png图片
如果图片是256*256,那么embedding是长度为512的向量。
import insightface
import cv2model = insightface.app.FaceAnalysis()
model.prepare(ctx_id=0, det_thresh=0.45)
face_img = cv2.imread('/home/wyw/tes_imgs/src_0.jpg')
res = model.get(face_img)
print('embedding: ', res[0].embedding)
print('embedding size: ', res[0].embedding.size)
更多推荐
insightface提取embedding
发布评论