|
|
@@ -0,0 +1,27 @@
|
|
|
+from paddlex import create_pipeline
|
|
|
+import time
|
|
|
+from pathlib import Path
|
|
|
+
|
|
|
+input_path = "./sample_data/300674-母公司现金流量表-扫描.png"
|
|
|
+
|
|
|
+pipeline_path = "./table_recognition_v2-zhch.yaml"
|
|
|
+pipeline_name = Path(pipeline_path).stem
|
|
|
+output_path = Path(f"./sample_data/single_pipeline_output/{pipeline_name}/")
|
|
|
+
|
|
|
+pipeline = create_pipeline(pipeline=pipeline_path)
|
|
|
+
|
|
|
+# For Image
|
|
|
+output = pipeline.predict(
|
|
|
+ input=input_path,
|
|
|
+ device="gpu", # 或者 "gpu" 如果你有 GPU 支持
|
|
|
+ use_doc_orientation_classify=True, # 开启文档方向分类
|
|
|
+ use_doc_unwarping=False, # 开启文档去畸变
|
|
|
+)
|
|
|
+
|
|
|
+# 可视化结果并保存 json 结果
|
|
|
+for res in output:
|
|
|
+ res.print()
|
|
|
+ # res.save_to_json(save_path="sample_data/output")
|
|
|
+ # res.save_to_markdown(save_path="sample_data/output")
|
|
|
+ output_path.mkdir(parents=True, exist_ok=True)
|
|
|
+ res.save_all(save_path=output_path.as_posix()) # 保存所有结果到指定路径
|
