Jetson Linux API Reference
32.7.5 Release
Public Member Functions
TRT_Context Class Reference

Detailed Description

Definition at line 54 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, bool isOnnxModel=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()

TRT_Context::~TRT_Context ( )

Member Function Documentation

◆ buildTrtContext()

void TRT_Context::buildTrtContext ( const string &  deployfile,
const string &  modelfile,
bool  bUseCPUBuf = false,
bool  isOnnxModel = false 
)

◆ destroyTrtContext()

void TRT_Context::destroyTrtContext ( bool  bUseCPUBuf = false)

◆ doInference()

void TRT_Context::doInference ( queue< vector< cv::Rect > > *  rectList_queue,
float *  input = NULL 
)

◆ getBatchSize()

uint32_t TRT_Context::getBatchSize ( ) const

◆ getBuffer()

void*& TRT_Context::getBuffer ( const int &  index)

◆ getChannel()

int TRT_Context::getChannel ( ) const

◆ getFilterNum()

int TRT_Context::getFilterNum ( ) const

◆ getInputBuf()

float*& TRT_Context::getInputBuf ( )

◆ getModelClassCnt()

int TRT_Context::getModelClassCnt ( ) const

◆ getNetHeight()

int TRT_Context::getNetHeight ( ) const

◆ getNetWidth()

int TRT_Context::getNetWidth ( ) const

◆ getNumTrtInstances()

uint32_t TRT_Context::getNumTrtInstances ( ) const

◆ getOffsets()

void* TRT_Context::getOffsets ( ) const

◆ getScales()

void* TRT_Context::getScales ( ) const

◆ setBatchSize()

void TRT_Context::setBatchSize ( const uint32_t &  batchsize)

◆ setDumpResult()

void TRT_Context::setDumpResult ( const bool &  dump_result)

◆ setFilterNum()

void TRT_Context::setFilterNum ( const unsigned int &  filter_num)

◆ setMode()

void TRT_Context::setMode ( const int &  mode)

◆ setModelIndex()

void TRT_Context::setModelIndex ( int  modelIndex)

◆ setTrtProfilerEnabled()

void TRT_Context::setTrtProfilerEnabled ( const bool &  enable_trt_profiler)

Field Documentation

◆ bbox_output_scales

float TRT_Context::bbox_output_scales[4]

Definition at line 159 of file trt_inference.h.

◆ classCnt

const int TRT_Context::classCnt

Definition at line 150 of file trt_inference.h.

◆ INPUT_BLOB_NAME

const char* TRT_Context::INPUT_BLOB_NAME

Definition at line 152 of file trt_inference.h.

◆ input_scale

float TRT_Context::input_scale[3]

Definition at line 158 of file trt_inference.h.

◆ offsets

int TRT_Context::offsets[3]

Definition at line 157 of file trt_inference.h.

◆ OUTPUT_BBOX_NAME

const char* TRT_Context::OUTPUT_BBOX_NAME

Definition at line 154 of file trt_inference.h.

◆ OUTPUT_BLOB_NAME

const char* TRT_Context::OUTPUT_BLOB_NAME

Definition at line 153 of file trt_inference.h.

◆ ParseFunc_ID

const int TRT_Context::ParseFunc_ID

Definition at line 160 of file trt_inference.h.

◆ STRIDE

const int TRT_Context::STRIDE

Definition at line 155 of file trt_inference.h.

◆ THRESHOLD

float TRT_Context::THRESHOLD[3]

Definition at line 151 of file trt_inference.h.

◆ WORKSPACE_SIZE

const int TRT_Context::WORKSPACE_SIZE

Definition at line 156 of file trt_inference.h.

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