DetectionOutput 插件层通过执行非极大值抑制，基于位置和置信度预测生成检测输出。更多...

#include <NvInferPluginUtils.h>

公共属性
bool	shareLocation
	如果为 true，则不同类别之间共享边界框。更多...

bool	varianceEncodedInTarget

int32_t	backgroundLabelId
	背景标签 ID。如果没有背景类别，请将其设置为 -1。更多...

int32_t	numClasses
	要预测的类别数量。更多...

int32_t	topK

int32_t	keepTopK
	在 NMS 步骤之后，每张图像要保留的边界框总数。更多...

float	confidenceThreshold
	仅考虑置信度大于阈值的检测结果。更多...

float	nmsThreshold
	在 NMS 中使用的阈值。更多...

CodeTypeSSD	codeType
	bbox 的编码方法类型。更多...

int32_t	inputOrder [3]
	指定输入的顺序 {loc_data, conf_data, priorbox_data}。更多...

bool	confSigmoid
	设置为 true 以计算置信度分数的 sigmoid。更多...

bool	isNormalized
	如果边界框数据已由网络标准化，则设置为 true。更多...

bool	isBatchAgnostic {true}
	默认为 true。如果先验框在每个批次中都是唯一的，则设置为 false。更多...

详细描述

DetectionOutput 插件层通过执行非极大值抑制，基于位置和置信度预测生成检测输出。

此插件首先基于生成的锚框解码边界框。然后，它对解码后的边界框执行非极大值抑制。 DetectionOutputParameters 定义了一组用于创建 DetectionOutput 插件层的参数。

已弃用: 在 TensorRT 10.0 中已弃用。 DetectionOutput 插件已被弃用。

成员数据文档

◆ backgroundLabelId

int32_t nvinfer1::plugin::DetectionOutputParameters::backgroundLabelId

背景标签 ID。如果没有背景类别，请将其设置为 -1。

◆ codeType

CodeTypeSSD nvinfer1::plugin::DetectionOutputParameters::codeType

bbox 的编码方法类型。

◆ confidenceThreshold

float nvinfer1::plugin::DetectionOutputParameters::confidenceThreshold

仅考虑置信度大于阈值的检测结果。

◆ confSigmoid

bool nvinfer1::plugin::DetectionOutputParameters::confSigmoid

设置为 true 以计算置信度分数的 sigmoid。

◆ inputOrder

int32_t nvinfer1::plugin::DetectionOutputParameters::inputOrder[3]

指定输入的顺序 {loc_data, conf_data, priorbox_data}。

◆ isBatchAgnostic

bool nvinfer1::plugin::DetectionOutputParameters::isBatchAgnostic {true}

默认为 true。如果先验框在每个批次中都是唯一的，则设置为 false。

◆ isNormalized

bool nvinfer1::plugin::DetectionOutputParameters::isNormalized

如果边界框数据已由网络标准化，则设置为 true。

◆ keepTopK

int32_t nvinfer1::plugin::DetectionOutputParameters::keepTopK

在 NMS 步骤之后，每张图像要保留的边界框总数。

◆ nmsThreshold

float nvinfer1::plugin::DetectionOutputParameters::nmsThreshold

在 NMS 中使用的阈值。

◆ numClasses

int32_t nvinfer1::plugin::DetectionOutputParameters::numClasses

要预测的类别数量。

◆ shareLocation

bool nvinfer1::plugin::DetectionOutputParameters::shareLocation

如果为 true，则不同类别之间共享边界框。

◆ topK

int32_t nvinfer1::plugin::DetectionOutputParameters::topK

每张图像中具有最高置信度分数并被馈送到 NMS 算法的框的数量。

◆ varianceEncodedInTarget

bool nvinfer1::plugin::DetectionOutputParameters::varianceEncodedInTarget

如果为 true，则方差在目标中编码。否则，我们需要相应地调整预测的偏移量。

此结构体的文档由以下文件生成

NvInferPluginUtils.h

公共属性