TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
•All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
Public Attributes | List of all members
SampleUffSSDParams Struct Reference

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

Inheritance diagram for SampleUffSSDParams:
Inheritance graph
[legend]
Collaboration diagram for SampleUffSSDParams:
Collaboration graph
[legend]

Public Attributes

std::string uffFileName
 The file name of the UFF model to use. More...
 
std::string labelsFileName
 The file namefo the class labels. More...
 
int32_t outputClsSize
 The number of output classes. More...
 
int32_t calBatchSize
 The size of calibration batch. More...
 
int32_t nbCalBatches
 The number of batches for calibration. More...
 
int32_t keepTopK
 The maximum number of detection post-NMS. More...
 
float visualThreshold
 The minimum score threshold to consider a detection. 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 SampleUffSSDParams structure groups the additional parameters required by the Uff SSD sample.

Member Data Documentation

◆ uffFileName

std::string SampleUffSSDParams::uffFileName

The file name of the UFF model to use.

◆ labelsFileName

std::string SampleUffSSDParams::labelsFileName

The file namefo the class labels.

◆ outputClsSize

int32_t SampleUffSSDParams::outputClsSize

The number of output classes.

◆ calBatchSize

int32_t SampleUffSSDParams::calBatchSize

The size of calibration batch.

◆ nbCalBatches

int32_t SampleUffSSDParams::nbCalBatches

The number of batches for calibration.

◆ keepTopK

int32_t SampleUffSSDParams::keepTopK

The maximum number of detection post-NMS.

◆ visualThreshold

float SampleUffSSDParams::visualThreshold

The minimum score threshold to consider a detection.

◆ 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: