TensorRT
6.0.1.5
|
RPROIParams is used to create the RPROIPlugin instance. It contains: More...
#include <NvInferPluginUtils.h>
Public Attributes | |
int | poolingH |
int | poolingW |
int | featureStride |
int | preNmsTop |
int | nmsMaxOut |
int | anchorsRatioCount |
int | anchorsScaleCount |
float | iouThreshold |
float | minBoxSize |
float | spatialScale |
RPROIParams is used to create the RPROIPlugin instance. It contains:
poolingH | Height of the output in pixels after ROI pooling on feature map. |
poolingW | Width of the output in pixels after ROI pooling on feature map. |
featureStride | Feature stride; ratio of input image size to feature map size. Assuming that max pooling layers in neural network use square filters. |
preNmsTop | Number of proposals to keep before applying NMS. |
nmsMaxOut | Number of remaining proposals after applying NMS. |
anchorsRatioCount | Number of anchor box ratios. |
anchorsScaleCount | Number of anchor box scales. |
iouThreshold | IoU (Intersection over Union) threshold used for the NMS step. |
minBoxSize | Minimum allowed bounding box size before scaling, used for anchor box calculation. |
spatialScale | Spatial scale between the input image and the last feature map. |