药材检测输出识别结果转换成类名，调整摄像头顺序

bsw215583320

2025-04-10 2052071ffeb03831056afea0b1810847fcec53b3

药材检测输出识别结果转换成类名，调整摄像头顺序

 config/herb_ai.yaml

@@ -1,10 +1,8 @@
win:
  className: "aimodel"
  windowName: "aimodel"

tag_win:
  className:
  #windowName: "recv"
  windowName: "recv"
log:
  path: "./log/herb_ai.log"
@@ -12,10 +10,9 @@
  safe: './model/safety_det.onnx'
  cls: './model/herb_identify.onnx'
cam:
  cam1: 0
  cam2: 1
  cam1: 1
  cam2: 0
  sleep: 0.1
  frames: 10
  # 新增清理策略配置
  days_threshold: 7      # 保留文件天数
  max_files: 100         # 最大文件数量
  days_threshold: 7
  max_files: 100

 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: