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