Ver código fonte

feat: 更新默认配置文件路径,添加新的示例配置以支持特定数据集

zhch158_admin 1 dia atrás
pai
commit
e4e1d1a927
1 arquivos alterados com 16 adições e 8 exclusões
  1. 16 8
      merger/merge_paddleocr_vl_paddleocr.py

+ 16 - 8
merger/merge_paddleocr_vl_paddleocr.py

@@ -284,6 +284,14 @@ if __name__ == "__main__":
     
     if len(sys.argv) == 1:
         # 默认配置
+        default_config = {
+            "paddleocr-vl-file": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水/paddleocr_vl_results/A用户_单元格扫描流水_page_001.json",
+            "paddle-file": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水/ppstructurev3_client_results/A用户_单元格扫描流水_page_001.json",
+            "output-dir": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水/paddleocr_vl_results_cell_bbox",
+            "output-type": "both",
+            "window": "15",
+            "threshold": "85"
+        }
         # 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",
@@ -292,14 +300,14 @@ if __name__ == "__main__":
         #     "window": "15",
         #     "threshold": "85"
         # }
-        default_config = {
-            "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/流水分析/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",