Definition at line 52 of file trt_inference.h.
TRT_Context::TRT_Context |
( |
| ) |
|
TRT_Context::~TRT_Context |
( |
| ) |
|
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: