XZNSH-Code-AI/xznsh_flow/yolov8_det.py

def analysis_yolov8(frame, model_coco, confidence):
    # 第一步：用COCO数据集推理
    results_coco = model_coco(frame)
    if not results_coco:
        return []
    boxes = results_coco[0].boxes
    result_list = []
    for box in boxes:
        # 过滤置信度0.5以下目标
        if float(box.conf) < confidence:
            continue
        labels_name = model_coco.names[int(box.cls)]
        label_xyxy_dict = {labels_name: box.xyxy[0].tolist()}
        result_list.append(label_xyxy_dict)
    return result_list