XZNSH-Code-AI/Bank_second_part/dataset_deal/yolov8_det.py

def analysis_yolov8(frame, model_coco,confidence):

    # 第一步：用COCO数据集推理
    results_coco = model_coco(frame)
    # print('results_coco:',results_coco)
    if results_coco:

        for r in results_coco:

            boxes = r.boxes

        re_dict_all = {}

        re_list = []
        idx = 0
        for box in boxes:

            idx += 1

            b = box.xyxy[0]  # get box coordinates in (top, left, bottom, right) format
            c = box.cls

            # 保存标签和坐标值作为返回结果
            blist = b.tolist()
            labels_name = model_coco.names[int(c)]
 
            confidence = float(box.conf)
                    
            confidence = round(confidence, 2)

            # 过滤置信度0.5以下目标
            if confidence < confidence:

                continue
            
            # 一个结果字典
            re_dict = [labels_name,blist]

            re_list.append(re_dict)

            # key = f"detection{idx}"
            # re_dict_all[key] = re_dict

    return re_list