NvInferPluginUtils.h 文件参考

#include "NvInferRuntimeCommon.h"

跳转到此文件的源代码。

类
struct	nvinfer1::plugin::PriorBoxParameters
	PriorBox 插件层在所有维度 (H x W) 上生成指定大小和宽高比的先验框。更多...
struct	nvinfer1::plugin::RPROIParams
	RPROIParams 用于创建 RPROIPlugin 实例。更多...
struct	nvinfer1::plugin::GridAnchorParameters
	Anchor Generator 插件层在所有维度 (H x W) 上生成指定大小和宽高比的先验框。GridAnchorParameters 定义了一组参数，用于为所有特征图创建插件层。更多...
struct	nvinfer1::plugin::DetectionOutputParameters
	DetectionOutput 插件层通过执行非极大值抑制，基于位置和置信度预测生成检测输出。更多...
struct	nvinfer1::plugin::softmaxTree
	当执行 yolo9000 时，softmaxTree 帮助对置信度分数进行 softmax 运算，以便通过词树结构化分类定义，为元素获得精确的分类。更多...
struct	nvinfer1::plugin::RegionParameters
	Region 插件层执行区域提议计算。更多...
struct	nvinfer1::plugin::NMSParameters
	NMSParameters 被 BatchedNMSPlugin 用于对目标检测网络的框执行非极大值抑制操作。更多...

命名空间
namespace	nvinfer1
	TensorRT API 版本 1 的命名空间。
namespace	nvinfer1::plugin

枚举
enum class	nvinfer1::plugin::CodeTypeSSD : int32_t { nvinfer1::plugin::CORNER = 0 , nvinfer1::plugin::CENTER_SIZE = 1 , nvinfer1::plugin::CORNER_SIZE = 2 , nvinfer1::plugin::TF_CENTER = 3 }
	用于解码边界框和 loc_data 的编码类型。更多...

详细描述

这是 Nvidia 提供的 TensorRT 插件实用程序的 API。它列出了 TensorRT 插件使用的所有参数。