forked from open-mmlab/mmdetection3d
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Feature] better tensorrt cpp code (open-mmlab#11)
* better tensorrt cpp code * fix end of file
- Loading branch information
Showing
27 changed files
with
573 additions
and
1,043 deletions.
There are no files selected for viewing
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
3 changes: 1 addition & 2 deletions
3
...sorrt/batched_nms/batchedNMSInference.cpp → ...rt/batched_nms/trt_batched_nms_kernel.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
16 changes: 16 additions & 0 deletions
16
backend_ops/tensorrt/batched_nms/trt_batched_nms_kernel.hpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
#ifndef TRT_BATCHED_NMS_KERNEL_HPP | ||
#define TRT_BATCHED_NMS_KERNEL_HPP | ||
#include "cuda_runtime_api.h" | ||
#include "kernel.h" | ||
|
||
pluginStatus_t nmsInference( | ||
cudaStream_t stream, const int N, const int perBatchBoxesSize, | ||
const int perBatchScoresSize, const bool shareLocation, | ||
const int backgroundLabelId, const int numPredsPerClass, | ||
const int numClasses, const int topK, const int keepTopK, | ||
const float scoreThreshold, const float iouThreshold, | ||
const DataType DT_BBOX, const void* locData, const DataType DT_SCORE, | ||
const void* confData, void* nmsedDets, void* nmsedLabels, void* workspace, | ||
bool isNormalized, bool confSigmoid, bool clipBoxes); | ||
|
||
#endif |
Oops, something went wrong.