herb_ai.py
@@ -199,7 +199,7 @@ # 安全检测 boxes, scores, class_ids = safety_detect(frame) draw_img = safety_detect.draw_detections(frame, boxes, scores, class_ids) print(boxes, scores, class_ids) det_res = {} if class_ids is not None: # 遍历class_ids 转换成类别名称 @@ -211,7 +211,7 @@ det_res[class_name] = det_res[class_name] if det_res[class_name] > scores[i] else scores[i] else: det_res[class_name] = scores[i] print(det_res) logger.info(f"安全检测识别结果, {det_res}") # 如果cass_ids中包含0，则表示有安全检测到人体 if 0 in class_ids:
