From 0043848a52e7ca6c272d4e48bb069a5d8721e73a Mon Sep 17 00:00:00 2001 From: lzhangzz Date: Tue, 16 Nov 2021 15:23:07 +0800 Subject: [PATCH] fix TRTBatchedNMS (#204) Co-authored-by: Li Zhang --- backend_ops/tensorrt/batched_nms/trt_batched_nms.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend_ops/tensorrt/batched_nms/trt_batched_nms.cpp b/backend_ops/tensorrt/batched_nms/trt_batched_nms.cpp index 9256f49b3..49f50ed61 100644 --- a/backend_ops/tensorrt/batched_nms/trt_batched_nms.cpp +++ b/backend_ops/tensorrt/batched_nms/trt_batched_nms.cpp @@ -95,7 +95,7 @@ int TRTBatchedNMS::enqueue(const nvinfer1::PluginTensorDesc* inputDesc, bool shareLocation = (inputDesc[0].dims.d[2] == 1); int topk = param.topK > 0 && param.topK <= inputDesc[1].dims.d[1] - ? topk + ? param.topK : inputDesc[1].dims.d[1]; pluginStatus_t status = nmsInference( stream, batch_size, boxes_size, score_size, shareLocation,