TensorRT 10.0.1
nvinfer1::plugin::RegionParameters Struct Reference

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...
 
softmaxTreesmTree
 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:

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact