Browse Source

feat: 更新默认配置文件路径,调整示例文件以支持新的数据集

zhch158_admin 13 giờ trước cách đây
mục cha
commit
04cce2f558
1 tập tin đã thay đổi với 19 bổ sung3 xóa
  1. 19 3
      merger/merge_paddleocr_vl_paddleocr.py

+ 19 - 3
merger/merge_paddleocr_vl_paddleocr.py

@@ -284,15 +284,31 @@ if __name__ == "__main__":
     
     if len(sys.argv) == 1:
         # 默认配置
+        # default_config = {
+        #     "paddleocr-vl-file": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/paddleocr_vl_results/B用户_扫描流水_page_001.json",
+        #     "paddle-file": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/ppstructurev3_client_results/B用户_扫描流水_page_001.json",
+        #     "output-dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/paddleocr_vl_results_cell_bbox",
+        #     "output-type": "both",
+        #     "window": "15",
+        #     "threshold": "85"
+        # }
         default_config = {
-            "paddleocr-vl-file": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/paddleocr_vl_results/2023年度报告母公司_page_007.json",
-            "paddle-file": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/ppstructurev3_client_results/2023年度报告母公司_page_007.json",
+            "paddleocr-vl-file": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/paddleocr_vl_results/2023年度报告母公司_page_006.json",
+            "paddle-file": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/ppstructurev3_client_results/2023年度报告母公司_page_006.json",
             "output-dir": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/paddleocr_vl_results_cell_bbox",
             "output-type": "both",
             "window": "15",
             "threshold": "85"
         }
-        
+        # default_config = {
+        #     "paddleocr-vl-file": "/Users/zhch158/workspace/data/流水分析/康强_北京农村商业银行/paddleocr_vl_results/康强_北京农村商业银行_page_001.json",
+        #     "paddle-file": "/Users/zhch158/workspace/data/流水分析/康强_北京农村商业银行/ppstructurev3_client_results/康强_北京农村商业银行_page_001.json",
+        #     "output-dir": "/Users/zhch158/workspace/data/流水分析/康强_北京农村商业银行/paddleocr_vl_results_cell_bbox",
+        #     "output-type": "both",
+        #     "window": "15",
+        #     "threshold": "85"
+        # }
+            
         print("ℹ️  未提供命令行参数,使用默认配置运行...")
         print("⚙️  默认参数:")
         for key, value in default_config.items():