Просмотр исходного кода

bugfix:

1. add `border=1` to table markdown html code;
2. set use_doc_orientation_classify and use_doc_unwarping to False by
defalut.
gaotingquan 6 месяцев назад
Родитель
Сommit
6a6ea4886c

+ 2 - 2
paddlex/inference/pipelines/layout_parsing/pipeline_v2.py

@@ -950,8 +950,8 @@ class _LayoutParsingPipelineV2(BasePipeline):
     def predict(
         self,
         input: Union[str, list[str], np.ndarray, list[np.ndarray]],
-        use_doc_orientation_classify: Union[bool, None] = None,
-        use_doc_unwarping: Union[bool, None] = None,
+        use_doc_orientation_classify: Union[bool, None] = False,
+        use_doc_unwarping: Union[bool, None] = False,
         use_textline_orientation: Optional[bool] = None,
         use_seal_recognition: Union[bool, None] = None,
         use_table_recognition: Union[bool, None] = None,

+ 3 - 1
paddlex/inference/pipelines/layout_parsing/result_v2.py

@@ -429,7 +429,9 @@ class LayoutParsingResultV2(BaseCVResult, HtmlMixin, XlsxMixin, MarkdownMixin):
 
         if self["model_settings"].get("use_table_recognition", False):
             if pretty:
-                format_table_func = lambda block: "\n" + format_text_func(block)
+                format_table_func = lambda block: "\n" + format_text_func(
+                    block
+                ).replace("<table>", '<table border="1">')
             else:
                 format_table_func = lambda block: simplify_table_func(
                     "\n" + block.content