Jelajahi Sumber

fix ocr pipeline typo (#2897)

cuicheng01 10 bulan lalu
induk
melakukan
47fd00f0f2

+ 1 - 1
paddlex/configs/pipelines/OCR.yaml

@@ -41,4 +41,4 @@ SubModules:
     model_name: PP-OCRv4_mobile_rec 
     model_dir: null
     batch_size: 1
-    score_thresh: 0
+    score_thresh: 0.0

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

@@ -394,7 +394,7 @@ class OCRPipeline(BasePipeline):
                     if rec_res["rec_score"] >= text_rec_score_thresh:
                         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])
+                        single_img_res["rec_polys"].append(dt_polys[sno])
             if self.text_type == "general":
                 rec_boxes = convert_points_to_boxes(single_img_res["rec_polys"])
                 single_img_res["rec_boxes"] = rec_boxes