L4T Multimedia API Reference

28.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 *& getBuffer (const int &index)
 
float *& getInputBuf ()
 
uint32_t getNumTrtInstances () const
 
void setForcedFp32 (const bool &forced_fp32)
 
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::setDumpResult ( const bool &  dump_result)
void TRT_Context::setFilterNum ( const unsigned int &  filter_num)
void TRT_Context::setForcedFp32 ( const bool &  forced_fp32)
void TRT_Context::setModelIndex ( int  modelIndex)
void TRT_Context::setTrtProfilerEnabled ( const bool &  enable_trt_profiler)

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