Parcourir la source

feat: update input and output configurations for bank statement processing, adding support for new document types and adjusting configuration file references for improved processing accuracy

zhch158_admin il y a 6 jours
Parent
commit
01920d7a82
1 fichiers modifiés avec 7 ajouts et 3 suppressions
  1. 7 3
      zhch/universal_doc_parser/main_v2.py

+ 7 - 3
zhch/universal_doc_parser/main_v2.py

@@ -299,11 +299,15 @@ if __name__ == "__main__":
             # "input": "/Users/zhch158/workspace/data/流水分析/康强_北京农村商业银行.pdf",
             # "output_dir": "./output/康强_北京农村商业银行_bank_statement_v2",
 
-            "input": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/mineru_vllm_results/2023年度报告母公司/2023年度报告母公司_page_003.png",
-            "output_dir": "./output/2023年度报告母公司_bank_statement_v2",
+            # "input": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/mineru_vllm_results/2023年度报告母公司/2023年度报告母公司_page_003.png",
+            # "output_dir": "./output/2023年度报告母公司_bank_statement_v2",
             
+            "input": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水.pdf",
+            "output_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/bank_statement_yusys_v2",
+
             # 配置文件
-            "config": "./config/bank_statement_v2.yaml",
+            "config": "./config/bank_statement_yusys_v2.yaml",
+            # "config": "./config/bank_statement_v2.yaml",
             # "config": "./config/bank_statement_paddle_vl.yaml",
             
             # 场景