主机环境
代码下载
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
原图: