Ver Fonte

support print parsing_res_list

zhouchangda há 3 semanas atrás
pai
commit
eaa32c14d4
1 ficheiros alterados com 10 adições e 0 exclusões
  1. 10 0
      paddlex/inference/pipelines/paddleocr_vl/result.py

+ 10 - 0
paddlex/inference/pipelines/paddleocr_vl/result.py

@@ -324,6 +324,16 @@ class PaddleOCRVLResult(BaseCVResult, HtmlMixin, XlsxMixin, MarkdownMixin):
             data["doc_preprocessor_res"] = self["doc_preprocessor_res"].str["res"]
         if self["model_settings"]["use_layout_detection"]:
             data["layout_det_res"] = self["layout_det_res"].str["res"]
+        parsing_res_list = self["parsing_res_list"]
+        parsing_res_list = [
+            {
+                "block_label": parsing_res.label,
+                "block_content": parsing_res.content,
+                "block_bbox": parsing_res.bbox,
+            }
+            for parsing_res in parsing_res_list
+        ]
+        data["parsing_res_list"] = parsing_res_list
         return JsonMixin._to_str(data, *args, **kwargs)
 
     def _to_json(self, *args, **kwargs) -> dict[str, str]: