Переглянути джерело

refactor(main_v2): update output directory paths and configuration for YUSYS v3

- Changed output directory names from 'bank_statement_wired_unet' to 'bank_statement_yusys_v3' across various test cases and configurations.
- Updated the configuration file path to use 'bank_statement_yusys_v3.yaml' for consistency with the new version.
- Ensured alignment of logging and output settings with the updated directory structure.
zhch158_admin 1 місяць тому
батько
коміт
d001ea34cf
1 змінених файлів з 14 додано та 14 видалено
  1. 14 14
      ocr_tools/universal_doc_parser/main_v2.py

+ 14 - 14
ocr_tools/universal_doc_parser/main_v2.py

@@ -418,7 +418,7 @@ if __name__ == "__main__":
             # "output_dir": "./output/康强_北京农村商业银行/bank_statement_mineru_vl",
 
             # "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/A用户_单元格扫描流水_page_002.png",
-            # "output_dir": "./output/A用户_单元格扫描流水_bank_statement_wired_unet",
+            # "output_dir": "./output/A用户_单元格扫描流水_bank_statement_yusys_v3",
             
             # "input": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水.pdf",
             # "output_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/bank_statement_yusys_v2",
@@ -427,43 +427,43 @@ if __name__ == "__main__":
             # "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/2023年度报告母公司_page_003.png",
             # "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/2023年度报告母公司_page_003_270_skew(-0.4).png",
             # "input": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司.pdf",
-            # "output_dir": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/bank_statement_wired_unet",
+            # "output_dir": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/bank_statement_yusys_v3",
 
             # "input": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司.pdf",
             # "output_dir": "/Users/zhch158/workspace/data/流水分析/2023年度报告母公司/bank_statement_yusys_v2",
 
             # # "input": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水.pdf",
-            # "output_dir": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水/bank_statement_wired_unet",
+            # "output_dir": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水/bank_statement_yusys_v3",
 
             # "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/600916_中国黄金_2022年报_page_096.png",
-            # "output_dir": "./output/600916_中国黄金_2022年报/bank_statement_wired_unet",
+            # "output_dir": "./output/600916_中国黄金_2022年报/bank_statement_yusys_v3",
             # "input": "/Users/zhch158/workspace/data/流水分析/600916_中国黄金_2022年报.pdf",
-            # "output_dir": "./output/600916_中国黄金_2022年报/bank_statement_wired_unet",
+            # "output_dir": "./output/600916_中国黄金_2022年报/bank_statement_yusys_v3",
 
 
             # "input": "/Users/zhch158/workspace/data/流水分析/德_内蒙古银行照.pdf",
-            # "output_dir": "./output/德_内蒙古银行照/bank_statement_wired_unet",
+            # "output_dir": "./output/德_内蒙古银行照/bank_statement_yusys_v3",
 
             "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/提取自赤峰黄金2023年报.pdf",
-            "output_dir": "./output/提取自赤峰黄金2023年报/bank_statement_wired_unet",
+            "output_dir": "./output/提取自赤峰黄金2023年报/bank_statement_yusys_v3",
             # "input": "/Users/zhch158/workspace/data/流水分析/提取自赤峰黄金2023年报.pdf",
-            # "output_dir": "/Users/zhch158/workspace/data/流水分析/提取自赤峰黄金2023年报/bank_statement_wired_unet",
+            # "output_dir": "/Users/zhch158/workspace/data/流水分析/提取自赤峰黄金2023年报/bank_statement_yusys_v3",
 
             # "input": "/Users/zhch158/workspace/data/流水分析/施博深.pdf",
-            # "output_dir": "/Users/zhch158/workspace/data/流水分析/施博深/bank_statement_wired_unet",
+            # "output_dir": "/Users/zhch158/workspace/data/流水分析/施博深/bank_statement_yusys_v3",
             # "output_dir": "./output/施博深/bank_statement_smart_router",
 
             # "input": "/Users/zhch158/workspace/data/流水分析/施博深.wiredtable/施博深_page_020.png",
-            # "output_dir": "./output/施博深/bank_statement_wired_unet",
+            # "output_dir": "./output/施博深/bank_statement_yusys_v3",
 
             # "input": "/Users/zhch158/workspace/data/流水分析/施博深.wiredtable",
-            # "output_dir": "/Users/zhch158/workspace/data/流水分析/施博深/bank_statement_wired_unet",
+            # "output_dir": "/Users/zhch158/workspace/data/流水分析/施博深/bank_statement_yusys_v3",
 
             # "input": "/Users/zhch158/workspace/data/流水分析/山西云集科技有限公司.pdf",
-            # "output_dir": "/Users/zhch158/workspace/data/流水分析/山西云集科技有限公司/bank_statement_wired_unet",
+            # "output_dir": "/Users/zhch158/workspace/data/流水分析/山西云集科技有限公司/bank_statement_yusys_v3",
 
             # 配置文件
-            "config": "./config/bank_statement_wired_unet.yaml",
+            "config": "./config/bank_statement_yusys_v3.yaml",
             # "config": "./config/bank_statement_smart_router.yaml",
             # "config": "./config/bank_statement_mineru_vl.yaml",
             # "config": "./config/bank_statement_yusys_v2.yaml",
@@ -486,7 +486,7 @@ if __name__ == "__main__":
             "log_level": "DEBUG",
 
             # 日志文件
-            "log_file": "./output/logs/bank_statement_wired_unet/process.log",
+            "log_file": "./output/logs/bank_statement_yusys_v3/process.log",
         }
         
         # 构造参数