mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

delete debug

pull/5902/head
LDOUBLEV 2022-04-07 08:31:37 +00:00
parent a8c8f485cb
commit 5efa071c07
2 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ppocr/postprocess/rec_postprocess.py
View File

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

6
tools/infer/utility.py
View File

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