From 8587b40c923f5ce11c3fd1745184d7de740a42ee Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Thu, 19 Aug 2021 12:23:05 +0000 Subject: [PATCH] fix predict_e2e.py --- tools/infer/predict_e2e.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_e2e.py b/tools/infer/predict_e2e.py index cd6c2005..5029d605 100755 --- a/tools/infer/predict_e2e.py +++ b/tools/infer/predict_e2e.py @@ -74,7 +74,7 @@ class TextE2E(object): self.preprocess_op = create_operators(pre_process_list) self.postprocess_op = build_post_process(postprocess_params) - self.predictor, self.input_tensor, self.output_tensors = utility.create_predictor( + self.predictor, self.input_tensor, self.output_tensors, _ = utility.create_predictor( args, 'e2e', logger) # paddle.jit.load(args.det_model_dir) # self.predictor.eval() -- GitLab