yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-01 10:19:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix rec prediction for trt7

Browse Source
This commit is contained in:
littletomatodonkey 2022-05-13 14:28:35 +00:00
parent bb77fcef6f
commit a3753aae50
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tools/infer/utility.py
View File

@ -272,9 +272,10 @@ def create_predictor(args, mode, logger):
if args.rec_algorithm not in ["CRNN", "SVTR_LCNet"]:
use_dynamic_shape = False
imgH = int(args.rec_image_shape.split(',')[-2])
min_input_shape = {"x": [1, 3, imgH, 10]}
max_input_shape = {"x": [args.rec_batch_num, 3, imgH, 2304]}
opt_input_shape = {"x": [args.rec_batch_num, 3, imgH, 320]}
min_input_shape = {"x": [1, 3, imgH, 10], }
max_input_shape = {"x": [args.rec_batch_num, 3, imgH, 2304], }
opt_input_shape = {"x": [args.rec_batch_num, 3, imgH, 320], }
config.exp_disable_tensorrt_ops(["elementwise_add"])
elif mode == "cls":
min_input_shape = {"x": [1, 3, 48, 10]}
max_input_shape = {"x": [args.rec_batch_num, 3, 48, 1024]}