Update infer_yolov3.py

This commit is contained in:
Jason
2022-09-15 14:01:41 +08:00
committed by GitHub
parent a6860d168f
commit 18fa2c83d5

View File

@@ -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