TRT_Context Class Reference

Detailed Description

Definition at line 51 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 156 of file trt_inference.h.

classCnt

const int TRT_Context::classCnt

Definition at line 147 of file trt_inference.h.

INPUT_BLOB_NAME

const char* TRT_Context::INPUT_BLOB_NAME

Definition at line 149 of file trt_inference.h.

input_scale

float TRT_Context::input_scale[3]

Definition at line 155 of file trt_inference.h.

offsets

int TRT_Context::offsets[3]

Definition at line 154 of file trt_inference.h.

OUTPUT_BBOX_NAME

const char* TRT_Context::OUTPUT_BBOX_NAME

Definition at line 151 of file trt_inference.h.

OUTPUT_BLOB_NAME

const char* TRT_Context::OUTPUT_BLOB_NAME

Definition at line 150 of file trt_inference.h.

ParseFunc_ID

const int TRT_Context::ParseFunc_ID

Definition at line 157 of file trt_inference.h.

STRIDE

const int TRT_Context::STRIDE

Definition at line 152 of file trt_inference.h.

THRESHOLD

float TRT_Context::THRESHOLD[3]

Definition at line 148 of file trt_inference.h.

WORKSPACE_SIZE

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:

• multimedia_api/ll_samples/samples/common/algorithm/trt/trt_inference.h