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
SampleFasterRCNNParams Struct Reference

The SampleFasterRCNNParams structure groups the additional parameters required by the FasterRCNN sample. More...

Inheritance diagram for SampleFasterRCNNParams:
Collaboration diagram for SampleFasterRCNNParams:

Public Attributes

int outputClsSize
 The number of output classes. More...
 
int nmsMaxOut
 The maximum number of detection post-NMS. 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 SampleFasterRCNNParams structure groups the additional parameters required by the FasterRCNN sample.

Member Data Documentation

◆ outputClsSize

int SampleFasterRCNNParams::outputClsSize

The number of output classes.

◆ nmsMaxOut

int SampleFasterRCNNParams::nmsMaxOut

The maximum number of detection post-NMS.

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