提交 5efa071c 编写于 作者: L LDOUBLEV

delete debug

上级 a8c8f485
...@@ -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)
......
...@@ -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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册