Update batchedNMS plugin to IPluginV2DynamicExt

Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
NVIDIA · Aug 6, 2020 · a6e50c6 · a6e50c6
1 parent 3891af8
commit a6e50c6
Show file tree

Hide file tree

Showing 4 changed files with 372 additions and 58 deletions.
diff --git a/plugin/InferPlugin.cpp b/plugin/InferPlugin.cpp
@@ -158,6 +158,7 @@ bool initLibNvInferPlugins(void* logger, const char* libNamespace)
 {
     initializePlugin<nvinfer1::plugin::BatchTilePluginCreator>(logger, libNamespace);
     initializePlugin<nvinfer1::plugin::BatchedNMSPluginCreator>(logger, libNamespace);
+    initializePlugin<nvinfer1::plugin::BatchedNMSDynamicPluginCreator>(logger, libNamespace);
     initializePlugin<nvinfer1::plugin::CoordConvACPluginCreator>(logger, libNamespace);
     initializePlugin<nvinfer1::plugin::CropAndResizePluginCreator>(logger, libNamespace);
     initializePlugin<nvinfer1::plugin::DetectionLayerPluginCreator>(logger, libNamespace);

diff --git a/plugin/README.md b/plugin/README.md
@@ -6,6 +6,7 @@
 |---|---|---|
 | [batchTilePlugin](batchTilePlugin) | BatchTilePlugin_TRT | 1 |
 | [batchedNMSPlugin](batchedNMSPlugin) | BatchedNMS_TRT | 1 |
+| [batchedNMSDynamicPlugin](batchedNMSPlugin) | BatchedNMSDynamic_TRT | 1 |
 | [bertQKVToContextPlugin](bertQKVToContextPlugin) | CustomQKVToContextPluginDynamic | 1 |
 | [coordConvACPlugin](coordConvACPlugin) | CoordConvAC | 1 |
 | [cropAndResizePlugin](cropAndResizePlugin) | CropAndResize | 1 |