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

fix TRTBatchedNMS (#204) 

Co-authored-by: Li Zhang <zhangli@sensetime.com>
pull/12/head
lzhangzz 2021-11-16 15:23:07 +08:00 committed by GitHub
parent 49dd1cf678
commit 0043848a52
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
backend_ops/tensorrt/batched_nms/trt_batched_nms.cpp
View File

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