别的onnx文件 about onnx-scrfd-flask HOT 6 CLOSED

detector = SCRFD(model_file='onnx/scrfd_34g.onnx')
bboxes, kpss = detector.detect(frame, 0.5, input_size = (frame.shape[0], frame.shape[1]))

import argparse
from tools.scrfd import *
import datetime
import cv2

frame=cv2.imread("./image008006.jpg")

detector = SCRFD(model_file='onnx/scrfd_34g.onnx')
bboxes, kpss = detector.detect(frame, 0.5, input_size = (frame.shape[0], frame.shape[1]))
for i in range(bboxes.shape[0]):
    bbox = bboxes[i]
    x1, y1, x2, y2, score = bbox.astype(np.int)
    cv2.rectangle(frame, (x1, y1), (x2, y2), (255, 0, 0), 2)
    if kpss is not None:
        kps = kpss[i]
        for kp in kps:
            kp = kp.astype(np.int)
            cv2.circle(frame, tuple(kp), 1, (0, 0, 255), 2)
ret, jpeg = cv2.imencode('.jpg', frame)
frame = jpeg.tobytes()

print(bboxes)