RefineDetectionWorkSpace Struct Reference

Collaboration diagram for RefineDetectionWorkSpace:

Public Member Functions
	RefineDetectionWorkSpace (const int batchSize, const int sampleCount, const RefineNMSParameters &param, const nvinfer1::DataType type)
	RefineDetectionWorkSpace ()=default

Public Attributes
nvinfer1::DimsHW	argMaxScoreDims
nvinfer1::DimsHW	argMaxBboxDims
nvinfer1::DimsHW	argMaxLabelDims
nvinfer1::DimsHW	sortClassScoreDims
nvinfer1::DimsHW	sortClassLabelDims
nvinfer1::DimsHW	sortClassSampleIdxDims
nvinfer1::Dims	sortClassValidCountDims = {1, {1, 0}, {nvinfer1::DimensionType::kINDEX}}
nvinfer1::DimsHW	sortClassPosDims
nvinfer1::DimsHW	sortNMSMarkDims
size_t	argMaxScoreOffset = 0
size_t	argMaxBboxOffset = 0
size_t	argMaxLabelOffset = 0
size_t	sortClassScoreOffset = 0
size_t	sortClassLabelOffset = 0
size_t	sortClassSampleIdxOffset = 0
size_t	sortClassValidCountOffset = 0
size_t	sortClassPosOffset = 0
size_t	sortNMSMarkOffset = 0
size_t	totalSize = 0

Constructor & Destructor Documentation

RefineDetectionWorkSpace() [1/2]

RefineDetectionWorkSpace::RefineDetectionWorkSpace	(	const int	batchSize,
		const int	sampleCount,
		const RefineNMSParameters &	param,
		const nvinfer1::DataType	type
	)

RefineDetectionWorkSpace() [2/2]

RefineDetectionWorkSpace::RefineDetectionWorkSpace ( )

default

Member Data Documentation

argMaxScoreDims

nvinfer1::DimsHW RefineDetectionWorkSpace::argMaxScoreDims

argMaxBboxDims

nvinfer1::DimsHW RefineDetectionWorkSpace::argMaxBboxDims

argMaxLabelDims

nvinfer1::DimsHW RefineDetectionWorkSpace::argMaxLabelDims

sortClassScoreDims

nvinfer1::DimsHW RefineDetectionWorkSpace::sortClassScoreDims

sortClassLabelDims

nvinfer1::DimsHW RefineDetectionWorkSpace::sortClassLabelDims

sortClassSampleIdxDims

nvinfer1::DimsHW RefineDetectionWorkSpace::sortClassSampleIdxDims

sortClassValidCountDims

nvinfer1::Dims RefineDetectionWorkSpace::sortClassValidCountDims = {1, {1, 0}, {nvinfer1::DimensionType::kINDEX}}

sortClassPosDims

nvinfer1::DimsHW RefineDetectionWorkSpace::sortClassPosDims

sortNMSMarkDims

nvinfer1::DimsHW RefineDetectionWorkSpace::sortNMSMarkDims

argMaxScoreOffset

size_t RefineDetectionWorkSpace::argMaxScoreOffset = 0

argMaxBboxOffset

size_t RefineDetectionWorkSpace::argMaxBboxOffset = 0

argMaxLabelOffset

size_t RefineDetectionWorkSpace::argMaxLabelOffset = 0

sortClassScoreOffset

size_t RefineDetectionWorkSpace::sortClassScoreOffset = 0

sortClassLabelOffset

size_t RefineDetectionWorkSpace::sortClassLabelOffset = 0

sortClassSampleIdxOffset

size_t RefineDetectionWorkSpace::sortClassSampleIdxOffset = 0

sortClassValidCountOffset

size_t RefineDetectionWorkSpace::sortClassValidCountOffset = 0

sortClassPosOffset

size_t RefineDetectionWorkSpace::sortClassPosOffset = 0

sortNMSMarkOffset

size_t RefineDetectionWorkSpace::sortNMSMarkOffset = 0

totalSize

size_t RefineDetectionWorkSpace::totalSize = 0

---

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

• maskRCNNKernels.h