From 20354397a063629bb8cd9c740b3f5473b3dbb02e Mon Sep 17 00:00:00 2001 From: felixhjh <852142024@qq.com> Date: Fri, 14 Jan 2022 10:14:09 +0000 Subject: [PATCH] change postprocess name --- examples/Pipeline/PaddleOCR/ocr/web_service.py | 4 ++-- python/paddle_serving_app/reader/ocr_reader.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/Pipeline/PaddleOCR/ocr/web_service.py b/examples/Pipeline/PaddleOCR/ocr/web_service.py index ed9d0c86..5cb1ada5 100644 --- a/examples/Pipeline/PaddleOCR/ocr/web_service.py +++ b/examples/Pipeline/PaddleOCR/ocr/web_service.py @@ -213,13 +213,13 @@ class RecOp(Op): res_list = [] if isinstance(fetch_data, dict): if len(fetch_data) > 0: - rec_batch_res = self.ocr_reader.postprocess( + rec_batch_res = self.ocr_reader.postprocess_ocrv2( fetch_data, with_score=True) for res in rec_batch_res: res_list.append(res[0]) elif isinstance(fetch_data, list): for one_batch in fetch_data: - one_batch_res = self.ocr_reader.postprocess( + one_batch_res = self.ocr_reader.postprocess_ocrv2( one_batch, with_score=True) for res in one_batch_res: res_list.append(res[0]) diff --git a/python/paddle_serving_app/reader/ocr_reader.py b/python/paddle_serving_app/reader/ocr_reader.py index 61b4ed06..3db07ddf 100644 --- a/python/paddle_serving_app/reader/ocr_reader.py +++ b/python/paddle_serving_app/reader/ocr_reader.py @@ -279,7 +279,7 @@ class OCRReader(object): return norm_img_batch[0] - def postprocess_old(self, outputs, with_score=False): + def postprocess(self, outputs, with_score=False): rec_res = [] rec_idx_lod = outputs["ctc_greedy_decoder_0.tmp_0.lod"] rec_idx_batch = outputs["ctc_greedy_decoder_0.tmp_0"] @@ -309,7 +309,7 @@ class OCRReader(object): rec_res.append([preds_text]) return rec_res - def postprocess(self, outputs, with_score=False): + def postprocess_ocrv2(self, outputs, with_score=False): preds = outputs["save_infer_model/scale_0.tmp_1"] try: preds = preds.numpy() -- GitLab