|
|
@@ -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
|