Переглянути джерело

set printoptions for numpy (#2993)

changdazhou 9 місяців тому
батько
коміт
e1b20332f2

+ 2 - 0
paddlex/inference/common/result/base_result.py

@@ -16,6 +16,7 @@ import inspect
 from pathlib import Path
 import time
 import random
+import numpy as np
 
 from ....utils import logging
 from .mixin import StrMixin, JsonMixin, ImgMixin
@@ -37,6 +38,7 @@ class BaseResult(dict, JsonMixin, StrMixin):
         self._save_funcs = []
         StrMixin.__init__(self)
         JsonMixin.__init__(self)
+        np.set_printoptions(threshold=1, edgeitems=1)
         self._rand_fn = None
 
     def save_all(self, save_path: str) -> None:

+ 5 - 3
paddlex/inference/pipelines/ocr/result.py

@@ -144,12 +144,14 @@ class OCRResult(BaseCVResult):
         data["text_det_params"] = self["text_det_params"]
         data["text_type"] = self["text_type"]
         if "textline_orientation_angles" in self:
-            data["textline_orientation_angles"] = self["textline_orientation_angles"]
+            data["textline_orientation_angles"] = np.array(
+                self["textline_orientation_angles"]
+            )
         data["text_rec_score_thresh"] = self["text_rec_score_thresh"]
         data["rec_texts"] = self["rec_texts"]
-        data["rec_scores"] = self["rec_scores"]
+        data["rec_scores"] = np.array(self["rec_scores"])
         data["rec_polys"] = self["rec_polys"]
-        data["rec_boxes"] = self["rec_boxes"]
+        data["rec_boxes"] = np.array(self["rec_boxes"])
 
         return JsonMixin._to_str(data, *args, **kwargs)