Region 插件层执行区域提议计算。更多...

#include <NvInferPluginUtils.h>

公共属性
int32_t	num
	每个网格单元预测的边界框数量。更多...

int32_t	coords
	边界框的坐标数量。更多...

int32_t	classes
	要预测的分类数量。更多...

softmaxTree *	smTree
	用于对置信度分数进行 softmax 运算的辅助结构。更多...

详细描述

Region 插件层执行区域提议计算。

每个单元格生成 5 个边界框（对于 yolo9000，每个单元格生成 3 个边界框）。对于每个框，计算其从 80 个预定义分类中检测到对象的概率（yolo9000 有 9418 个预定义分类，这些 9418 个项目组织为工作树结构）。RegionParameters 定义了一组用于创建 Region 插件层的参数。

成员数据文档

◆ classes

int32_t nvinfer1::plugin::RegionParameters::classes

要预测的分类数量。

◆ coords

int32_t nvinfer1::plugin::RegionParameters::coords

边界框的坐标数量。

◆ num

int32_t nvinfer1::plugin::RegionParameters::num

每个网格单元预测的边界框数量。

◆ smTree

softmaxTree* nvinfer1::plugin::RegionParameters::smTree

用于对置信度分数进行 softmax 运算的辅助结构。

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