Sfoglia il codice sorgente

Fix ocr pipeline & Support saving images in the dict (#2900)

* Fix ocr pipeline & Support saving images in the dict

* update
cuicheng01 10 mesi fa
parent
commit
0a07b0a42e

+ 1 - 2
paddlex/inference/pipelines_new/layout_parsing/result.py

@@ -76,8 +76,7 @@ class LayoutParsingResult(BaseCVResult, HtmlMixin, XlsxMixin):
                 formula_region_id = formula_res["formula_region_id"]
                 sub_formula_res_dict = formula_res.img
                 key = f"formula_res_region{formula_region_id}"
-                res_img_dict[key] = sub_formula_res_dict
-
+                res_img_dict[key] = sub_formula_res_dict["res"]
         return res_img_dict
 
     def _to_str(self, *args, **kwargs) -> Dict[str, str]:

+ 1 - 1
paddlex/inference/pipelines_new/ocr/pipeline.py

@@ -363,7 +363,7 @@ class OCRPipeline(BasePipeline):
                 # use textline orientation model
                 if model_settings["use_textline_orientation"]:
                     angles = [
-                        textline_angle_info["class_ids"][0]
+                        int(textline_angle_info["class_ids"][0])
                         for textline_angle_info in self.textline_orientation_model(
                             all_subs_of_img
                         )