瀏覽代碼

fix seal pp (#2867)

* fix seal pp

* Update pipeline.py
Sunflower7788 10 月之前
父節點
當前提交
163d2f9f8b
共有 1 個文件被更改,包括 6 次插入3 次删除
  1. 6 3
      paddlex/inference/pipelines_new/ocr/pipeline.py

+ 6 - 3
paddlex/inference/pipelines_new/ocr/pipeline.py

@@ -380,7 +380,10 @@ class OCRPipeline(BasePipeline):
                         single_img_res["rec_texts"].append(rec_res["rec_text"])
                         single_img_res["rec_scores"].append(rec_res["rec_score"])
                         single_img_res["rec_polys"].append(dt_polys[rno])
-
-            rec_boxes = convert_points_to_boxes(single_img_res["rec_polys"])
-            single_img_res["rec_boxes"] = rec_boxes
+            if self.text_type == "general":
+                rec_boxes = convert_points_to_boxes(single_img_res["rec_polys"])
+                single_img_res["rec_boxes"] = rec_boxes
+            else:
+                single_img_res["rec_boxes"] = np.array([])
+                
             yield OCRResult(single_img_res)