mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmdeploy/backend_ops/tensorrt/batched_nms
History
q.yao 5998d24766
[Feature] Add TensorRT batched NMS support (#3) 
* add trt batched_nms plugin

* update trt batched nms
 		2021-06-25 19:31:16 +08:00
..
allClassNMS.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
batchedNMSInference.cpp [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
cub_helper.h [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
gatherNMSOutputs.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
kernel.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
kernel.h [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
permuteData.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
sortScoresPerClass.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
sortScoresPerImage.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
trt_batched_nms.cpp [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
trt_batched_nms.hpp [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00