Definition at line 366 of file nvdsinfer_context_impl.h.
Public Member Functions | |
OtherPostprocessor (int id, int gpuId=0) | |
NvDsInferStatus | initResource (const NvDsInferContextInitParams &initParams) override |
void | setDlHandle (const std::shared_ptr< DlLibHandle > &dlHandle) |
void | setNetworkInfo (const NvDsInferNetworkInfo &info) |
void | setAllLayerInfo (std::vector< NvDsInferBatchDimsLayerInfo > &info) |
void | setOutputLayerInfo (std::vector< NvDsInferBatchDimsLayerInfo > &info) |
void | setLoggingFunc (const NvDsInferLoggingFunc &func) |
const std::vector< std::vector< std::string > > & | getLabels () const |
bool | needInputCopy () const |
bool | needOutputCopyB4Processing () const |
virtual NvDsInferStatus | copyBuffersToHostMemory (NvDsInferBatch &buffer, CudaStream &mainStream) |
virtual NvDsInferStatus | postProcessHost (NvDsInferBatch &buffer, NvDsInferContextBatchOutput &output) |
void | freeBatchOutput (NvDsInferContextBatchOutput &batchOutput) |
Protected Member Functions | |
NvDsInferStatus | parseLabelsFile (const std::string &path) |
NvDsInferStatus | allocDeviceResource () |
void | releaseFrameOutput (NvDsInferFrameOutput &frameOutput) |
Protected Attributes | |
NvDsInferNetworkType | m_NetworkType = NvDsInferNetworkType_Other |
int | m_UniqueID = 0 |
uint32_t | m_GpuID = 0 |
NvDsInferLoggingFunc | m_LoggingFunc |
std::shared_ptr< DlLibHandle > | m_CustomLibHandle |
bool | m_CopyInputToHostBuffers = false |
bool | m_disableOutputHostCopy = false |
bool | m_DumpOpTensor = false |
std::vector< std::pair< std::string, std::string > > | m_DumpOpTensorFiles |
bool | m_OverwriteOpTensor = false |
std::vector< std::pair< std::string, int > > | m_OverwriteOpTensorFilePairs |
std::vector< std::ifstream * > | m_OverwriteOpTensorFiles |
NvDsInferNetworkInfo | m_NetworkInfo = {0} |
std::vector< NvDsInferLayerInfo > | m_AllLayerInfo |
std::vector< NvDsInferLayerInfo > | m_OutputLayerInfo |
std::vector< std::vector< std::string > > | m_Labels |
|
inline |
Definition at line 369 of file nvdsinfer_context_impl.h.
|
protectedinherited |
|
virtualinherited |
|
inherited |
|
inlineinherited |
Definition at line 146 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_Labels.
|
overridevirtual |
Reimplemented from nvdsinfer::InferPostprocessor.
|
inlineinherited |
Definition at line 150 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_CopyInputToHostBuffers.
|
inlineinherited |
Definition at line 152 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_disableOutputHostCopy.
|
protectedinherited |
|
virtualinherited |
|
protectedinherited |
|
inlineinherited |
Definition at line 132 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_AllLayerInfo.
|
inlineinherited |
Definition at line 124 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_CustomLibHandle.
|
inlineinherited |
Definition at line 142 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_LoggingFunc.
|
inlineinherited |
Definition at line 128 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_NetworkInfo.
|
inlineinherited |
Definition at line 137 of file nvdsinfer_context_impl.h.
References nvdsinfer::InferPostprocessor::m_OutputLayerInfo.
|
protectedinherited |
Definition at line 199 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::setAllLayerInfo().
|
protectedinherited |
Definition at line 190 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::needInputCopy().
|
protectedinherited |
Definition at line 189 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::setDlHandle().
|
protectedinherited |
Definition at line 191 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::needOutputCopyB4Processing().
|
protectedinherited |
Definition at line 192 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 193 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 185 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 203 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::getLabels().
|
protectedinherited |
Definition at line 186 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::setLoggingFunc().
|
protectedinherited |
Definition at line 198 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::setNetworkInfo().
|
protectedinherited |
Definition at line 182 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 200 of file nvdsinfer_context_impl.h.
Referenced by nvdsinfer::InferPostprocessor::setOutputLayerInfo().
|
protectedinherited |
Definition at line 194 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 195 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 196 of file nvdsinfer_context_impl.h.
|
protectedinherited |
Definition at line 184 of file nvdsinfer_context_impl.h.