mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity
mmdeploy/csrc/backend_ops/tensorrt/common_impl/nms
History
q.yao e3a8baac4c
[Enhancement] Support RotatedRetinaNet TensorRT (#422) 
* add rotated nms trt plugin

* fix ops output shape

* rebase

* fix lint

* add fp16, benchmark

* format docs

* remove unused definition, add ut

* add docs

* update docs

* add doc
2022-05-24 10:34:22 +08:00
..
allClassNMS.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
allClassRotatedNMS.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
batched_nms_kernel.cpp
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
gatherNMSOutputs.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
kernel.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
permuteData.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
sortScoresPerClass.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00
sortScoresPerImage.cu
[Enhancement] Support RotatedRetinaNet TensorRT (#422)
2022-05-24 10:34:22 +08:00