Provides pre-processing functionality like mean subtraction and normalization.
Definition at line 69 of file nvdsinfer_context_impl.h.
Public Member Functions | |
InferPreprocessor (const NvDsInferNetworkInfo &info, NvDsInferFormat format, const NvDsInferBatchDimsLayerInfo &layerInfo, int id=0) | |
virtual | ~InferPreprocessor ()=default |
void | setLoggingFunc (const NvDsInferLoggingFunc &func) |
bool | setScaleOffsets (float scale, const std::vector< float > &offsets={}) |
bool | setMeanFile (const std::string &file) |
bool | setInputOrder (const NvDsInferTensorOrder order) |
NvDsInferStatus | allocateResource () |
NvDsInferStatus | syncStream () |
NvDsInferStatus | transform (NvDsInferContextBatchInput &batchInput, void *devBuf, CudaStream &mainStream, CudaEvent *waitingEvent) |
nvdsinfer::InferPreprocessor::InferPreprocessor | ( | const NvDsInferNetworkInfo & | info, |
NvDsInferFormat | format, | ||
const NvDsInferBatchDimsLayerInfo & | layerInfo, | ||
int | id = 0 |
||
) |
|
virtualdefault |
NvDsInferStatus nvdsinfer::InferPreprocessor::allocateResource | ( | ) |
bool nvdsinfer::InferPreprocessor::setInputOrder | ( | const NvDsInferTensorOrder | order | ) |
|
inline |
Definition at line 76 of file nvdsinfer_context_impl.h.
bool nvdsinfer::InferPreprocessor::setMeanFile | ( | const std::string & | file | ) |
bool nvdsinfer::InferPreprocessor::setScaleOffsets | ( | float | scale, |
const std::vector< float > & | offsets = {} |
||
) |
NvDsInferStatus nvdsinfer::InferPreprocessor::syncStream | ( | ) |
NvDsInferStatus nvdsinfer::InferPreprocessor::transform | ( | NvDsInferContextBatchInput & | batchInput, |
void * | devBuf, | ||
CudaStream & | mainStream, | ||
CudaEvent * | waitingEvent | ||
) |