Provides pre-processing functionality like mean subtraction and normalization.
Definition at line 183 of file nvdspreprocess_impl.h.
Public Member Functions | |
NvDsPreProcessTensorImpl (const NvDsPreProcessNetworkSize &size, NvDsPreProcessFormat format, int id=0) | |
constructor for tensor preparation implementation More... | |
virtual | ~NvDsPreProcessTensorImpl ()=default |
bool | setScaleOffsets (float scale, const std::vector< float > &offsets={}) |
method to set offsets values More... | |
bool | setMeanFile (const std::string &file) |
method to set mean file More... | |
bool | setInputOrder (const NvDsPreProcessNetworkInputOrder order) |
method to set network input order More... | |
NvDsPreProcessStatus | allocateResource () |
allocate resources for tensor preparation More... | |
NvDsPreProcessStatus | syncStream () |
synchronize cuda stream More... | |
NvDsPreProcessStatus | prepare_tensor (NvDsPreProcessBatch *batch, void *&devBuf) |
method to prepare tensor using cuda kernels More... | |
NvDsPreProcessTensorImpl::NvDsPreProcessTensorImpl | ( | const NvDsPreProcessNetworkSize & | size, |
NvDsPreProcessFormat | format, | ||
int | id = 0 |
||
) |
constructor for tensor preparation implementation
|
virtualdefault |
NvDsPreProcessStatus NvDsPreProcessTensorImpl::allocateResource | ( | ) |
allocate resources for tensor preparation
NvDsPreProcessStatus NvDsPreProcessTensorImpl::prepare_tensor | ( | NvDsPreProcessBatch * | batch, |
void *& | devBuf | ||
) |
method to prepare tensor using cuda kernels
bool NvDsPreProcessTensorImpl::setInputOrder | ( | const NvDsPreProcessNetworkInputOrder | order | ) |
method to set network input order
bool NvDsPreProcessTensorImpl::setMeanFile | ( | const std::string & | file | ) |
method to set mean file
bool NvDsPreProcessTensorImpl::setScaleOffsets | ( | float | scale, |
const std::vector< float > & | offsets = {} |
||
) |
method to set offsets values
NvDsPreProcessStatus NvDsPreProcessTensorImpl::syncStream | ( | ) |
synchronize cuda stream