瀏覽代碼

fix(config): 更新模型路径和OCR置信度阈值以提高识别准确性

zhch158_admin 2 周之前
父節點
當前提交
da38a1ac18

+ 1 - 1
ocr_tools/universal_doc_parser/config/bank_statement_paddle_vl.yaml

@@ -18,7 +18,7 @@ preprocessor:
 layout_detection:
   # module: "paddle"
   # model_name: "RT-DETR-H_layout_17cls"
-  # model_dir: /Users/zhch158/workspace/repository.git/PaddleX/zhch/unified_pytorch_models/Layout/RT-DETR-H_layout_17cls.onnx  # 使用默认路径,或指定: "./Layout/RT-DETR-H_layout_17cls.onnx"
+  # model_dir: /Users/zhch158/models/pytorch_models/Layout/RT-DETR-H_layout_17cls.onnx  # 使用默认路径,或指定: "./Layout/RT-DETR-H_layout_17cls.onnx"
   # batch_size: 4
   # conf: 0.1
   # iou: 0.45

+ 2 - 2
ocr_tools/universal_doc_parser/config/bank_statement_yusys_v3.yaml

@@ -68,7 +68,7 @@ table_recognition_wired:
   need_ocr: true
   row_threshold: 10
   col_threshold: 15
-  ocr_conf_threshold: 0.8       # 单元格 OCR 置信度阈值
+  ocr_conf_threshold: 0.9       # 单元格 OCR 置信度阈值
   cell_crop_margin: 2
   use_custom_postprocess: true  # 是否使用自定义后处理(默认启用)
 
@@ -81,7 +81,7 @@ table_recognition_wired:
   # 融合引擎配置
   cell_fusion:
     # RT-DETR 模型路径(必需)
-    rtdetr_model_path: "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/pytorch_models/Table/RT-DETR-L_wired_table_cell_det.onnx"
+    rtdetr_model_path: "/Users/zhch158/models/pytorch_models/Table/RT-DETR-L_wired_table_cell_det.onnx"
     
     # 融合权重
     unet_weight: 0.6        # UNet 权重(结构性强)