TensorRT：Nvinfer.h-下篇

最新推荐文章于 2024-06-25 13:03:34 发布

微风❤水墨

最新推荐文章于 2024-06-25 13:03:34 发布

阅读量772

点赞数

分类专栏：目标检测-实现

本文链接：https://blog.csdn.net/lilai619/article/details/108054779

版权

目标检测-实现专栏收录该内容

22 篇文章 90 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文深入探讨了TensorRT中的PluginV2接口，介绍了如何配合PluginCreator进行使用。同时，讲解了IPluginRegistry作为插件注册的全局点，以及INetworkDefinition、ICudaEngine、IBuilder类在构建和执行网络中的作用。此外，还提到了ILogger类在日志记录中的应用。

摘要由CSDN通过智能技术生成

9：PluginV2类

Pluginv2是新版本的插件接口，部分函数接口与V1是一致的

class IPluginV2
{
public:
    //获取TensorRT的版本，用于向后兼容
    //! \brief Return the API version with which this plugin was built.
    //!
    //! Do not override this method as it is used by the TensorRT library to maintain backwards-compatibility with plugins.
    //!
    virtual int getTensorRTVersion() const
    {
        return NV_TENSORRT_VERSION;
    }

    //获取插件的类型
    //! \brief Return the plugin type. Should match the plugin name returned by the corresponding plugin creator
    // \see IPluginCreator::getPluginName()
    //!
    virtual const char* getPluginType() const = 0;

    //获取插件的版本
    //! \brief Return the plugin version. Should match the plugin version returned b

了解本专栏