Skip to content

GitLab

"src/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "9fb25cccc9b2d957c598c15775b8452cdd923155"
Commit 5efa071c authored by LDOUBLEV's avatar LDOUBLEV
Browse files

delete debug

parent a8c8f485
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
ppocr/postprocess/rec_postprocess.py
View file @ 5efa071c
...@@ -93,7 +93,6 @@ class CTCLabelDecode(BaseRecLabelDecode): ...@@ -93,7 +93,6 @@ class CTCLabelDecode(BaseRecLabelDecode):
preds = preds[-1] preds = preds[-1]
if isinstance(preds, paddle.Tensor): if isinstance(preds, paddle.Tensor):
preds = preds.numpy() preds = preds.numpy()
print(preds.shape)
preds_idx = preds.argmax(axis=2) preds_idx = preds.argmax(axis=2)
preds_prob = preds.max(axis=2) preds_prob = preds.max(axis=2)
text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True) text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True)
......
tools/infer/utility.py
View file @ 5efa071c
...@@ -313,12 +313,6 @@ def create_predictor(args, mode, logger): ...@@ -313,12 +313,6 @@ def create_predictor(args, mode, logger):
for name in input_names: for name in input_names:
input_tensor = predictor.get_input_handle(name) input_tensor = predictor.get_input_handle(name)
output_tensors = get_output_tensors(args, mode, predictor) 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 return predictor, input_tensor, output_tensors, config
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment