https://github.com/deepinsight/insightface/tree/master/examples/person_detection
detector = insightface.model_zoo.get_model('scrfd_person_2.5g.onnx', download=True)
detector.prepare(0, nms_thresh=0.5, input_size=(640, 640))
pip install -U insightface
onnx下载地址:
http://storage.insightface.ai/files/models/scrfd_person_2.5g.onnx
测试demo,本人把相关代码提取出来,不用安装insightface了,cpu 640*640 50ms
from __future__ import division
# -*- coding: utf-8 -*-
import time
import datetime
import numpy as np
import onnx
import onnxruntime
import os
import os.path as osp
import cv2
de