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

add params min_subgraph_size

Browse Source
This commit is contained in:
LDOUBLEV 2021-06-10 14:29:56 +08:00
parent 96a53fb4be
commit ab0b1ace8f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/infer/utility.py
View File

@ -236,8 +236,8 @@ def create_predictor(args, mode, logger):
config.enable_tensorrt_engine(
precision_mode=inference.PrecisionType.Float32,
max_batch_size=args.max_batch_size,
min_subgraph_size=args.
min_subgraph_size) # skip the minmum trt subgraph
min_subgraph_size=args.min_subgraph_size)
# skip the minmum trt subgraph
if mode == "det":
min_input_shape = {
"x": [1, 3, 50, 50],