Provides pre-processing functionality like mean subtraction and normalization.
Definition at line 70 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 77 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 | ||
| ) |