mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix clip on tensorrt 8.0 (#1833)

pull/1863/head
q.yao 2023-03-08 17:24:56 +08:00 committed by GitHub
parent b59d83bf7b
commit 7f30e42026
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmdeploy/codebase/mmdet/deploy/utils.py
View File

@ -75,7 +75,7 @@ def clip_bboxes(x1: Tensor, y1: Tensor, x2: Tensor, y2: Tensor,
@FUNCTION_REWRITER.register_rewriter(
func_name='mmdeploy.codebase.mmdet.deploy.utils.clip_bboxes',
backend='tensorrt',
extra_checkers=LibVersionChecker('tensorrt', min_version='8'))
extra_checkers=LibVersionChecker('tensorrt', min_version='8.2'))
def clip_bboxes__trt8(x1: Tensor, y1: Tensor, x2: Tensor, y2: Tensor,
max_shape: Union[Tensor, Sequence[int]]):
"""Clip bboxes for onnx. From TensorRT 8 we can do the operators on the