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::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 | ) |
|
float TRT_Context::bbox_output_scales[4] |
const int TRT_Context::classCnt |
const char* TRT_Context::INPUT_BLOB_NAME |
float TRT_Context::input_scale[3] |
int TRT_Context::offsets[3] |
const char* TRT_Context::OUTPUT_BBOX_NAME |
const char* TRT_Context::OUTPUT_BLOB_NAME |
const int TRT_Context::ParseFunc_ID |
const int TRT_Context::STRIDE |
float TRT_Context::THRESHOLD[3] |
const int TRT_Context::WORKSPACE_SIZE |
The documentation for this class was generated from the following file: