TensorRT 8.6.0
nvinfer1::plugin::RegionParameters Struct Reference

The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for yolo9000, generate 3 bounding boxes per cell). For each box, calculating its probablities of objects detections from 80 pre-defined classifications (yolo9000 has 9418 pre-defined classifications, and these 9418 items are organized as work-tree structure). RegionParameters defines a set of parameters for creating the Region plugin layer. More...

#include <NvInferPluginUtils.h>

Public Attributes

int32_t num
 
int32_t coords
 
int32_t classes
 
softmaxTreesmTree
 

Detailed Description

The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for yolo9000, generate 3 bounding boxes per cell). For each box, calculating its probablities of objects detections from 80 pre-defined classifications (yolo9000 has 9418 pre-defined classifications, and these 9418 items are organized as work-tree structure). RegionParameters defines a set of parameters for creating the Region plugin layer.

Parameters
numNumber of predicted bounding box for each grid cell.
coordsNumber of coordinates for a bounding box.
classesNumber of classifications to be predicted.
smTreeHelping structure to do softmax on confidence scores.

Member Data Documentation

◆ 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

The documentation for this struct was generated from the following file: