TensorRT  6.0.1.5
nvinfer1::plugin::softmaxTree 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

int * leaf
 
int n
 
int * parent
 
int * child
 
int * group
 
char ** name
 
int groups
 
int * groupSize
 
int * groupOffset
 

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 classfications to be predicted.
softmaxTreeWhen performing yolo9000, softmaxTree is helping to do softmax on confidence scores, for element to get the precise classfication through word-tree structured classfication definition.
Deprecated:
. This plugin is superseded by createRegionPlugin()

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