From 5efa071c075a0423920d2a2470d560b3035abd3a Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 7 Apr 2022 08:31:37 +0000 Subject: [PATCH] delete debug --- ppocr/postprocess/rec_postprocess.py | 1 - tools/infer/utility.py | 6 ------ 2 files changed, 7 deletions(-) diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index 0ffe917b..3bc7bcdf 100644 --- a/ppocr/postprocess/rec_postprocess.py +++ b/ppocr/postprocess/rec_postprocess.py @@ -93,7 +93,6 @@ class CTCLabelDecode(BaseRecLabelDecode): preds = preds[-1] if isinstance(preds, paddle.Tensor): preds = preds.numpy() - print(preds.shape) preds_idx = preds.argmax(axis=2) preds_prob = preds.max(axis=2) text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 1b282a01..fd2605d9 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -313,12 +313,6 @@ def create_predictor(args, mode, logger): for name in input_names: input_tensor = predictor.get_input_handle(name) output_tensors = get_output_tensors(args, mode, predictor) - #output_names = predictor.get_output_names() - #output_tensors = [] - #print("output names", output_names) - #for output_name in output_names: - # output_tensor = predictor.get_output_handle(output_name) - # output_tensors.append(output_tensor) return predictor, input_tensor, output_tensors, config -- GitLab