SampleSSDParams Struct Reference

The SampleSSDParams structure groups the additional parameters required by the SSD sample. More...

Inheritance diagram for SampleSSDParams:

Collaboration diagram for SampleSSDParams:

Public Attributes
int	outputClsSize
	The number of output classes. More...
int	keepTopK
	The maximum number of detection post-NMS. More...
int	nbCalBatches
	The number of batches for calibration. More...
float	visualThreshold
	The minimum score threshold to consider a detection. More...
std::string	calibrationBatches
	The path to calibration batches. More...
std::string	prototxtFileName
	Filename of prototxt design file of a network. More...
std::string	weightsFileName
	Filename of trained weights file of a network. More...
std::string	meanFileName
	Filename of mean file of a network. More...
int32_t	batchSize {1}
	Number of inputs in a batch. More...
int32_t	dlaCore {-1}
	Specify the DLA core to run network on. More...
bool	int8 {false}
	Allow runnning the network in Int8 mode. More...
bool	fp16 {false}
	Allow running the network in FP16 mode. More...
std::vector< std::string >	dataDirs
	Directory paths where sample data files are stored. More...
std::vector< std::string >	inputTensorNames
std::vector< std::string >	outputTensorNames

Detailed Description

The SampleSSDParams structure groups the additional parameters required by the SSD sample.

Member Data Documentation

outputClsSize

int SampleSSDParams::outputClsSize

The number of output classes.

keepTopK

int SampleSSDParams::keepTopK

The maximum number of detection post-NMS.

nbCalBatches

int SampleSSDParams::nbCalBatches

The number of batches for calibration.

visualThreshold

float SampleSSDParams::visualThreshold

The minimum score threshold to consider a detection.

calibrationBatches

std::string SampleSSDParams::calibrationBatches

The path to calibration batches.

prototxtFileName

std::string samplesCommon::CaffeSampleParams::prototxtFileName

inherited

Filename of prototxt design file of a network.

weightsFileName

std::string samplesCommon::CaffeSampleParams::weightsFileName

inherited

Filename of trained weights file of a network.

meanFileName

std::string samplesCommon::CaffeSampleParams::meanFileName

inherited

Filename of mean file of a network.

batchSize

int32_t samplesCommon::SampleParams::batchSize {1}

inherited

Number of inputs in a batch.

dlaCore

int32_t samplesCommon::SampleParams::dlaCore {-1}

inherited

Specify the DLA core to run network on.

int8

bool samplesCommon::SampleParams::int8 {false}

inherited

Allow runnning the network in Int8 mode.

fp16

bool samplesCommon::SampleParams::fp16 {false}

inherited

Allow running the network in FP16 mode.

dataDirs

std::vector<std::string> samplesCommon::SampleParams::dataDirs

inherited

Directory paths where sample data files are stored.

inputTensorNames

std::vector<std::string> samplesCommon::SampleParams::inputTensorNames

inherited

outputTensorNames

std::vector<std::string> samplesCommon::SampleParams::outputTensorNames

inherited

---

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

• sampleSSD.cpp