Update docs

This commit is contained in:
felixhjh
2023-02-08 16:33:12 +00:00
parent bc3a7ce8fe
commit 844d80ecae
61 changed files with 1117 additions and 2519 deletions

View File

@@ -32,6 +32,10 @@ def build_option(args):
if args.use_trt:
option.use_trt_backend()
# If use original Tensorrt, not Paddle-TensorRT,
# comment the following two lines
option.enable_paddle_to_trt()
option.enable_paddle_trt_collect_shape()
option.set_trt_input_shape("x", [1, 3, 256, 256], [1, 3, 1024, 1024],
[1, 3, 2048, 2048])
return option