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>
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
-
num | Number of predicted bounding box for each grid cell. |
coords | Number of coordinates for a bounding box. |
classes | Number of classifications to be predicted. |
smTree | Helping structure to do softmax on confidence scores. |
◆ 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: