0
点赞
收藏
分享

微信扫一扫

Insightface人脸特征识别模型环境搭建和试玩儿

谁知我新 2022-05-03 阅读 31

主机环境

代码下载

git clone https://github.com/deepinsight/insightface.git

 insightface/python-package at master · deepinsight/insightface · GitHub

环境准备

pip install -U Cython cmake numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install -U insightface -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

测试推理代码

按照GITHUB上提供的测试推理代码:

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)

执行python py.py

得到输出推理结果t1_output.jpg

原图:


结束

举报

相关推荐

0 条评论