|
|
@@ -169,7 +169,12 @@ def process_single_input(
|
|
|
}
|
|
|
else:
|
|
|
# 批量处理模式（原有逻辑）
|
|
|
- results = pipeline.process_document(str(input_path), page_range=page_range)
|
|
|
+ # 批量处理模式（原有逻辑）
|
|
|
+ results = pipeline.process_document(
|
|
|
+ str(input_path),
|
|
|
+ page_range=page_range,
|
|
|
+ output_dir=str(output_dir)
|
|
|
+ )
|
|
|
process_time = (datetime.now() - start_time).total_seconds()
|
|
|
|
|
|
logger.info(f"⏱️ 处理耗时: {process_time:.2f}秒")
|
|
|
@@ -402,8 +407,8 @@ if __name__ == "__main__":
|
|
|
# "input": "/Users/zhch158/workspace/data/流水分析/康强_北京农村商业银行.pdf",
|
|
|
# "output_dir": "./output/康强_北京农村商业银行_bank_statement_v2",
|
|
|
|
|
|
- "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",
|
|
|
+ # "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",
|
|
|
|
|
|
# "input": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水.pdf",
|
|
|
# "output_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/bank_statement_yusys_v2",
|
|
|
@@ -411,12 +416,15 @@ if __name__ == "__main__":
|
|
|
# "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/2023年度报告母公司_page_005.png",
|
|
|
# "input": "/Users/zhch158/workspace/repository.git/ocr_platform/ocr_tools/universal_doc_parser/tests/2023年度报告母公司_page_003_270.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": "./output/2023年度报告母公司/bank_statement_wired_unet",
|
|
|
|
|
|
# "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_v2",
|
|
|
|
|
|
+ "input": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水.pdf",
|
|
|
+ "output_dir": "/Users/zhch158/workspace/data/流水分析/A用户_单元格扫描流水/bank_statement_wired_unet",
|
|
|
+
|
|
|
# "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",
|
|
|
# "input": "/Users/zhch158/workspace/data/流水分析/600916_中国黄金_2022年报.pdf",
|
