diff --git a/plugin/batchedNMSPlugin/batchedNMSPlugin.cpp b/plugin/batchedNMSPlugin/batchedNMSPlugin.cpp index 8ba1596c..17582233 100644 --- a/plugin/batchedNMSPlugin/batchedNMSPlugin.cpp +++ b/plugin/batchedNMSPlugin/batchedNMSPlugin.cpp @@ -187,12 +187,12 @@ IPluginV2Ext* BatchedNMSPlugin::clone() const void BatchedNMSPlugin::setPluginNamespace(const char* pluginNamespace) { - mPluginNamespace = pluginNamespace; + mNamespace = pluginNamespace; } const char* BatchedNMSPlugin::getPluginNamespace() const { - return mPluginNamespace; + return mNamespace; } nvinfer1::DataType BatchedNMSPlugin::getOutputDataType( diff --git a/plugin/batchedNMSPlugin/batchedNMSPlugin.h b/plugin/batchedNMSPlugin/batchedNMSPlugin.h index 4e63d8c6..69322b08 100644 --- a/plugin/batchedNMSPlugin/batchedNMSPlugin.h +++ b/plugin/batchedNMSPlugin/batchedNMSPlugin.h @@ -88,7 +88,6 @@ class BatchedNMSPlugin : public IPluginV2Ext int numPriors{}; std::string mNamespace; bool mClipBoxes{}; - const char* mPluginNamespace; }; class BatchedNMSPluginCreator : public BaseCreator