mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 00:33:03 +08:00
Update infer_picodet.py
This commit is contained in:
@@ -34,8 +34,6 @@ def build_option(args):
|
||||
|
||||
if args.use_trt:
|
||||
option.use_trt_backend()
|
||||
option.set_trt_input_shape("image", [1, 3, 320, 320])
|
||||
option.set_trt_input_shape("scale_factor", [1, 2])
|
||||
return option
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user