The Region plugin layer performs region proposal calculation. More...

#include <NvInferPluginUtils.h>

Public Attributes
int32_t	num
	Number of predicted bounding box for each grid cell. More...

int32_t	coords
	Number of coordinates for a bounding box. More...

int32_t	classes
	Number of classifications to be predicted. More...

softmaxTree *	smTree
	Helping structure to do softmax on confidence scores. More...

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 probabilities 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.

Member Data Documentation

◆ classes

int32_t nvinfer1::plugin::RegionParameters::classes

Number of classifications to be predicted.

◆ coords

int32_t nvinfer1::plugin::RegionParameters::coords

Number of coordinates for a bounding box.

◆ num

int32_t nvinfer1::plugin::RegionParameters::num

Number of predicted bounding box for each grid cell.

◆ smTree

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

Helping structure to do softmax on confidence scores.

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