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:
|
if args.use_trt:
|
||||||
option.use_trt_backend()
|
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
|
return option
|
||||||
|
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user