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:

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