Sfoglia il codice sorgente

add pretty_markdown to config

gaotingquan 6 mesi fa
parent
commit
24953c6ac2

+ 1 - 0
paddlex/configs/pipelines/PP-StructureV3.yaml

@@ -6,6 +6,7 @@ use_general_ocr: True
 use_seal_recognition: True
 use_table_recognition: True
 use_formula_recognition: True
+pretty_markdown: True
 
 SubModules:
   LayoutDetection:

+ 8 - 0
paddlex/inference/pipelines/layout_parsing/pipeline_v2.py

@@ -114,6 +114,11 @@ class LayoutParsingPipelineV2(BasePipeline):
             True,
         )
 
+        self.pretty_markdown = config.get(
+            "pretty_markdown",
+            True,
+        )
+
         if self.use_doc_preprocessor:
             doc_preprocessor_config = config.get("SubPipelines", {}).get(
                 "DocPreprocessor",
@@ -939,6 +944,9 @@ class LayoutParsingPipelineV2(BasePipeline):
         if use_chart_recognition is None:
             use_chart_recognition = self.use_chart_recognition
 
+        if pretty_markdown is None:
+            pretty_markdown = self.pretty_markdown
+
         return dict(
             use_doc_preprocessor=use_doc_preprocessor,
             use_general_ocr=use_general_ocr,