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