infer_preprocess_kernel.h File Reference

Go to the source code of this file.

Data Structures
struct	BatchSurfaceInfo

Macros
#define	NVDSINFER_MAX_PREPROC_CHANNELS   4
	Header file for pre-processing CUDA kernels with normalization and mean subtraction required by nvdsinferserver. More...
#define	NVDSINFER_INVALID_CONVERSION_KEY   UINT32_MAX

Typedefs
typedef void(*	NvDsInferConvertFcn) (void *outBuffer, int outDataType, BatchSurfaceInfo outBufferInfo, void *inBuffer, int inDatatype, BatchSurfaceInfo inBufferInfo, int validN, int cropC, int cropH, int cropW, float scaleFactor, float *meanDataBuffer, int fromChannelIndices[NVDSINFER_MAX_PREPROC_CHANNELS], int toNCHW, cudaStream_t stream)
	Function pointer type to which any of the NvDsInferConvert functions can be assigned. More...

Functions
bool	NvDsInferConvert_CxToPx (void *outBuffer, int outDataType, BatchSurfaceInfo outBufferInfo, void *inBuffer, int inDatatype, BatchSurfaceInfo inBufferInfo, int validN, int cropC, int cropH, int cropW, float scaleFactor, float meanOffsets[NVDSINFER_MAX_PREPROC_CHANNELS], int fromChannelIndices[NVDSINFER_MAX_PREPROC_CHANNELS], int toNCHW, cudaStream_t stream)
bool	NvDsInferConvert_CxToPxWithMeanBuffer (void *outBuffer, int outDataType, BatchSurfaceInfo outBufferInfo, void *inBuffer, int inDatatype, BatchSurfaceInfo inBufferInfo, int validN, int cropC, int cropH, int cropW, float scaleFactor, float *meanDataBuffer, int fromChannelIndices[NVDSINFER_MAX_PREPROC_CHANNELS], int toNCHW, cudaStream_t stream)

Macro Definition Documentation

NVDSINFER_INVALID_CONVERSION_KEY

#define NVDSINFER_INVALID_CONVERSION_KEY   UINT32_MAX

Definition at line 25 of file infer_preprocess_kernel.h.

NVDSINFER_MAX_PREPROC_CHANNELS

#define NVDSINFER_MAX_PREPROC_CHANNELS   4

Header file for pre-processing CUDA kernels with normalization and mean subtraction required by nvdsinferserver.

Definition at line 24 of file infer_preprocess_kernel.h.

Typedef Documentation

NvDsInferConvertFcn

typedef void(* NvDsInferConvertFcn) (void *outBuffer, int outDataType, BatchSurfaceInfo outBufferInfo, void *inBuffer, int inDatatype, BatchSurfaceInfo inBufferInfo, int validN, int cropC, int cropH, int cropW, float scaleFactor, float *meanDataBuffer, int fromChannelIndices[NVDSINFER_MAX_PREPROC_CHANNELS], int toNCHW, cudaStream_t stream)

Function pointer type to which any of the NvDsInferConvert functions can be assigned.

Definition at line 57 of file infer_preprocess_kernel.h.

Function Documentation

NvDsInferConvert_CxToPx()

bool NvDsInferConvert_CxToPx	(	void *	outBuffer,
		int	outDataType,
		BatchSurfaceInfo	outBufferInfo,
		void *	inBuffer,
		int	inDatatype,
		BatchSurfaceInfo	inBufferInfo,
		int	validN,
		int	cropC,
		int	cropH,
		int	cropW,
		float	scaleFactor,
		float	meanOffsets[NVDSINFER_MAX_PREPROC_CHANNELS],
		int	fromChannelIndices[NVDSINFER_MAX_PREPROC_CHANNELS],
		int	toNCHW,
		cudaStream_t	stream
	)

NvDsInferConvert_CxToPxWithMeanBuffer()

bool NvDsInferConvert_CxToPxWithMeanBuffer	(	void *	outBuffer,
		int	outDataType,
		BatchSurfaceInfo	outBufferInfo,
		void *	inBuffer,
		int	inDatatype,
		BatchSurfaceInfo	inBufferInfo,
		int	validN,
		int	cropC,
		int	cropH,
		int	cropW,
		float	scaleFactor,
		float *	meanDataBuffer,
		int	fromChannelIndices[NVDSINFER_MAX_PREPROC_CHANNELS],
		int	toNCHW,
		cudaStream_t	stream
	)