L4T Multimedia API Reference

32.3.1 Release

 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
TRT_Context Class Reference

Detailed Description

Definition at line 52 of file trt_inference.h.

Public Member Functions

int getNetWidth () const
 
int getNetHeight () const
 
uint32_t getBatchSize () const
 
int getChannel () const
 
int getModelClassCnt () const
 
void * getScales () const
 
void * getOffsets () const
 
void *& getBuffer (const int &index)
 
float *& getInputBuf ()
 
uint32_t getNumTrtInstances () const
 
void setMode (const int &mode)
 
void setBatchSize (const uint32_t &batchsize)
 
void setDumpResult (const bool &dump_result)
 
void setTrtProfilerEnabled (const bool &enable_trt_profiler)
 
int getFilterNum () const
 
void setFilterNum (const unsigned int &filter_num)
 
 TRT_Context ()
 
void setModelIndex (int modelIndex)
 
void buildTrtContext (const string &deployfile, const string &modelfile, bool bUseCPUBuf=false)
 
void doInference (queue< vector< cv::Rect > > *rectList_queue, float *input=NULL)
 
void destroyTrtContext (bool bUseCPUBuf=false)
 
 ~TRT_Context ()
 

Constructor & Destructor Documentation

TRT_Context::TRT_Context ( )
TRT_Context::~TRT_Context ( )

Member Function Documentation

void TRT_Context::buildTrtContext ( const string &  deployfile,
const string &  modelfile,
bool  bUseCPUBuf = false 
)
void TRT_Context::destroyTrtContext ( bool  bUseCPUBuf = false)
void TRT_Context::doInference ( queue< vector< cv::Rect > > *  rectList_queue,
float *  input = NULL 
)
uint32_t TRT_Context::getBatchSize ( ) const
void*& TRT_Context::getBuffer ( const int &  index)
int TRT_Context::getChannel ( ) const
int TRT_Context::getFilterNum ( ) const
float*& TRT_Context::getInputBuf ( )
int TRT_Context::getModelClassCnt ( ) const
int TRT_Context::getNetHeight ( ) const
int TRT_Context::getNetWidth ( ) const
uint32_t TRT_Context::getNumTrtInstances ( ) const
void* TRT_Context::getOffsets ( ) const
void* TRT_Context::getScales ( ) const
void TRT_Context::setBatchSize ( const uint32_t &  batchsize)
void TRT_Context::setDumpResult ( const bool &  dump_result)
void TRT_Context::setFilterNum ( const unsigned int &  filter_num)
void TRT_Context::setMode ( const int &  mode)
void TRT_Context::setModelIndex ( int  modelIndex)
void TRT_Context::setTrtProfilerEnabled ( const bool &  enable_trt_profiler)

Field Documentation

float TRT_Context::bbox_output_scales[4]

Definition at line 156 of file trt_inference.h.

const int TRT_Context::classCnt

Definition at line 147 of file trt_inference.h.

const char* TRT_Context::INPUT_BLOB_NAME

Definition at line 149 of file trt_inference.h.

float TRT_Context::input_scale[3]

Definition at line 155 of file trt_inference.h.

int TRT_Context::offsets[3]

Definition at line 154 of file trt_inference.h.

const char* TRT_Context::OUTPUT_BBOX_NAME

Definition at line 151 of file trt_inference.h.

const char* TRT_Context::OUTPUT_BLOB_NAME

Definition at line 150 of file trt_inference.h.

const int TRT_Context::ParseFunc_ID

Definition at line 157 of file trt_inference.h.

const int TRT_Context::STRIDE

Definition at line 152 of file trt_inference.h.

float TRT_Context::THRESHOLD[3]

Definition at line 148 of file trt_inference.h.

const int TRT_Context::WORKSPACE_SIZE

Definition at line 153 of file trt_inference.h.


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