Update infer.py

This commit is contained in:
Jason
2022-09-15 14:02:39 +08:00
committed by GitHub
parent 9de96ec504
commit c65755493f

View File

@@ -34,8 +34,6 @@ def build_option(args):
if args.use_trt:
option.use_trt_backend()
option.set_trt_input_shape("inputs", [1, 3, 224, 224],
[1, 3, 224, 224], [1, 3, 224, 224])
return option