Detailed Description

A generic post processor class.

Definition at line 59 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Inheritance diagram for nvdsinferserver::Postprocessor:

[legend]

Collaboration diagram for nvdsinferserver::Postprocessor:

[legend]

Public Types
using	TensorAllocator = std::function< SharedSysMem(const std::string &name, size_t bytes)>

using	EventAllocator = std::function< SharedCuEvent()>

using	TensorAllocator = std::function< SharedSysMem(const std::string &name, size_t bytes)>

using	EventAllocator = std::function< SharedCuEvent()>

using	PostprocessDone = std::function< void(NvDsInferStatus, SharedBatchArray)>

using	PostprocessDone = std::function< void(NvDsInferStatus, SharedBatchArray)>

Public Member Functions
virtual	~Postprocessor ()=default

void	setDllHandle (const SharedDllHandle &dlHandle)

void	setLabelPath (const std::string &path)

void	setNetworkInfo (const NvDsInferNetworkInfo &info)

void	setOutputLayerCount (uint32_t num)

void	setInputCopy (bool enable)

const std::vector< std::vector< std::string > > &	getLabels () const

void	setAllocator (TensorAllocator cpuAlloc, EventAllocator event)

bool	needInputCopy () const

virtual NvDsInferStatus	allocateResource (const std::vector< int > &devIds) override

virtual	~Postprocessor ()=default

void	setDllHandle (const SharedDllHandle &dlHandle)

void	setLabelPath (const std::string &path)

void	setNetworkInfo (const NvDsInferNetworkInfo &info)

void	setOutputLayerCount (uint32_t num)

void	setInputCopy (bool enable)

const std::vector< std::vector< std::string > > &	getLabels () const

void	setAllocator (TensorAllocator cpuAlloc, EventAllocator event)

bool	needInputCopy () const

virtual NvDsInferStatus	allocateResource (const std::vector< int > &devIds) override

NvDsInferStatus	postCudaProcess (SharedBatchArray inBuf, SharedCuStream mainStream, PostprocessDone done) override
	Acquire an output buffer array and call CUDA post processing steps. More...

NvDsInferStatus	postCudaProcess (SharedBatchArray inBuf, SharedCuStream mainStream, PostprocessDone done) override
	Acquire an output buffer array and call CUDA post processing steps. More...

NvDsInferStatus	postHostProcess (SharedBatchArray inBuf, SharedCuStream mainStream, PostprocessDone done) override
	Acquire an output buffer array and call host side processing steps. More...

NvDsInferStatus	postHostProcess (SharedBatchArray inBuf, SharedCuStream mainStream, PostprocessDone done) override
	Acquire an output buffer array and call host side processing steps. More...

Protected Member Functions
	Postprocessor (InferPostprocessType type, int id)

NvDsInferStatus	postCudaImpl (SharedBatchArray &inBuf, SharedBatchArray &outbuf, SharedCuStream &mainStream) override

NvDsInferStatus	postHostImpl (SharedBatchArray &inBuf, SharedBatchArray &outbuf, SharedCuStream &mainStream) override

SharedBatchArray	requestCudaOutBufs (const SharedBatchArray &inBuf) override

SharedBatchArray	requestHostOutBufs (const SharedBatchArray &inBuf) override

NvDsInferStatus	parseLabelsFile (const std::string &path)

	Postprocessor (InferPostprocessType type, int id)

NvDsInferStatus	postCudaImpl (SharedBatchArray &inBuf, SharedBatchArray &outbuf, SharedCuStream &mainStream) override

NvDsInferStatus	postHostImpl (SharedBatchArray &inBuf, SharedBatchArray &outbuf, SharedCuStream &mainStream) override

SharedBatchArray	requestCudaOutBufs (const SharedBatchArray &inBuf) override

SharedBatchArray	requestHostOutBufs (const SharedBatchArray &inBuf) override

NvDsInferStatus	parseLabelsFile (const std::string &path)

Protected Attributes
SharedDllHandle	m_CustomLibHandle
	Custom library implementation. More...

bool	m_CopyInputToHostBuffers = false

NvDsInferNetworkInfo	m_NetworkInfo = {0}
	Network input information. More...

uint32_t	m_OutputLayerCount = 0

std::string	m_LabelPath

std::vector< std::vector< std::string > >	m_Labels
	Holds the string labels for classes. More...

TensorAllocator	m_CpuAllocator

EventAllocator	m_EventAllocator

Member Typedef Documentation

◆ EventAllocator [1/2]

using nvdsinferserver::Postprocessor::EventAllocator = std::function<SharedCuEvent()>

Definition at line 62 of file sources/libs/nvdsinferserver/infer_postprocess.h.

◆ EventAllocator [2/2]

using nvdsinferserver::Postprocessor::EventAllocator = std::function<SharedCuEvent()>

Definition at line 62 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

◆ PostprocessDone [1/2]

using nvdsinferserver::IPostprocessor::PostprocessDone = std::function<void(NvDsInferStatus, SharedBatchArray)>

inherited

Definition at line 67 of file sources/libs/nvdsinferserver/infer_iprocess.h.

◆ PostprocessDone [2/2]

using nvdsinferserver::IPostprocessor::PostprocessDone = std::function<void(NvDsInferStatus, SharedBatchArray)>

inherited

Definition at line 67 of file 9.1/sources/libs/nvdsinferserver/infer_iprocess.h.

◆ TensorAllocator [1/2]

using nvdsinferserver::Postprocessor::TensorAllocator = std::function<SharedSysMem(const std::string& name, size_t bytes)>

Definition at line 61 of file sources/libs/nvdsinferserver/infer_postprocess.h.

◆ TensorAllocator [2/2]

using nvdsinferserver::Postprocessor::TensorAllocator = std::function<SharedSysMem(const std::string& name, size_t bytes)>

Definition at line 61 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

Constructor & Destructor Documentation

◆ Postprocessor() [1/2]

nvdsinferserver::Postprocessor::Postprocessor	(	InferPostprocessType	type,
		int	id
	)

inlineprotected

Definition at line 65 of file sources/libs/nvdsinferserver/infer_postprocess.h.

◆ ~Postprocessor() [1/2]

virtual nvdsinferserver::Postprocessor::~Postprocessor ( )

virtualdefault

◆ Postprocessor() [2/2]

nvdsinferserver::Postprocessor::Postprocessor	(	InferPostprocessType	type,
		int	id
	)

inlineprotected

Definition at line 65 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

◆ ~Postprocessor() [2/2]

virtual nvdsinferserver::Postprocessor::~Postprocessor ( )

virtualdefault

Member Function Documentation

◆ allocateResource() [1/2]

virtual NvDsInferStatus nvdsinferserver::Postprocessor::allocateResource ( const std::vector< int > & devIds )

overridevirtual

Implements nvdsinferserver::BasePostprocessor.

Reimplemented in nvdsinferserver::SegmentPostprocessor, nvdsinferserver::SegmentPostprocessor, nvdsinferserver::ClassifyPostprocessor, nvdsinferserver::ClassifyPostprocessor, nvdsinferserver::DetectPostprocessor, and nvdsinferserver::DetectPostprocessor.

◆ allocateResource() [2/2]

virtual NvDsInferStatus nvdsinferserver::Postprocessor::allocateResource ( const std::vector< int > & devIds )

overridevirtual

Implements nvdsinferserver::BasePostprocessor.

Reimplemented in nvdsinferserver::SegmentPostprocessor, nvdsinferserver::SegmentPostprocessor, nvdsinferserver::ClassifyPostprocessor, nvdsinferserver::ClassifyPostprocessor, nvdsinferserver::DetectPostprocessor, and nvdsinferserver::DetectPostprocessor.

◆ getLabels() [1/2]

const std::vector<std::vector<std::string> >& nvdsinferserver::Postprocessor::getLabels ( ) const

inline

Definition at line 79 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_Labels.

◆ getLabels() [2/2]

const std::vector<std::vector<std::string> >& nvdsinferserver::Postprocessor::getLabels ( ) const

inline

Definition at line 79 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_Labels.

◆ needInputCopy() [1/2]

bool nvdsinferserver::Postprocessor::needInputCopy ( ) const

inline

Definition at line 87 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CopyInputToHostBuffers.

◆ needInputCopy() [2/2]

bool nvdsinferserver::Postprocessor::needInputCopy ( ) const

inline

Definition at line 87 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CopyInputToHostBuffers.

◆ networkType() [1/2]

InferPostprocessType nvdsinferserver::BasePostprocessor::networkType ( ) const

inlineinherited

Definition at line 185 of file sources/libs/nvdsinferserver/infer_iprocess.h.

◆ networkType() [2/2]

InferPostprocessType nvdsinferserver::BasePostprocessor::networkType ( ) const

inlineinherited

Definition at line 185 of file 9.1/sources/libs/nvdsinferserver/infer_iprocess.h.

◆ parseLabelsFile() [1/2]

NvDsInferStatus nvdsinferserver::Postprocessor::parseLabelsFile ( const std::string & path )

protected

◆ parseLabelsFile() [2/2]

NvDsInferStatus nvdsinferserver::Postprocessor::parseLabelsFile ( const std::string & path )

protected

◆ postCudaImpl() [1/2]

NvDsInferStatus nvdsinferserver::Postprocessor::postCudaImpl	(	SharedBatchArray &	inBuf,
		SharedBatchArray &	outbuf,
		SharedCuStream &	mainStream
	)

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ postCudaImpl() [2/2]

NvDsInferStatus nvdsinferserver::Postprocessor::postCudaImpl	(	SharedBatchArray &	inBuf,
		SharedBatchArray &	outbuf,
		SharedCuStream &	mainStream
	)

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ postCudaProcess() [1/2]

NvDsInferStatus nvdsinferserver::BasePostprocessor::postCudaProcess	(	SharedBatchArray	inBuf,
		SharedCuStream	mainStream,
		PostprocessDone	done
	)

inlineoverridevirtualinherited

Acquire an output buffer array and call CUDA post processing steps.

Parameters

[in]	inBuf	Pointer to the input batch array.
[in]	mainStream	The main CUDA stream to synchronize with.
[in]	done	Callback function to be executed.

Returns: Error status.

Implements nvdsinferserver::IPostprocessor.

Definition at line 195 of file sources/libs/nvdsinferserver/infer_iprocess.h.

◆ postCudaProcess() [2/2]

NvDsInferStatus nvdsinferserver::BasePostprocessor::postCudaProcess	(	SharedBatchArray	inBuf,
		SharedCuStream	mainStream,
		PostprocessDone	done
	)

inlineoverridevirtualinherited

Acquire an output buffer array and call CUDA post processing steps.

Parameters

[in]	inBuf	Pointer to the input batch array.
[in]	mainStream	The main CUDA stream to synchronize with.
[in]	done	Callback function to be executed.

Returns: Error status.

Implements nvdsinferserver::IPostprocessor.

Definition at line 195 of file 9.1/sources/libs/nvdsinferserver/infer_iprocess.h.

◆ postHostImpl() [1/2]

NvDsInferStatus nvdsinferserver::Postprocessor::postHostImpl	(	SharedBatchArray &	inBuf,
		SharedBatchArray &	outbuf,
		SharedCuStream &	mainStream
	)

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ postHostImpl() [2/2]

NvDsInferStatus nvdsinferserver::Postprocessor::postHostImpl	(	SharedBatchArray &	inBuf,
		SharedBatchArray &	outbuf,
		SharedCuStream &	mainStream
	)

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ postHostProcess() [1/2]

NvDsInferStatus nvdsinferserver::BasePostprocessor::postHostProcess	(	SharedBatchArray	inBuf,
		SharedCuStream	mainStream,
		PostprocessDone	done
	)

inlineoverridevirtualinherited

Acquire an output buffer array and call host side processing steps.

Parameters

[in]	inBuf	Pointer to the input batch array.
[in]	mainStream	The main CUDA stream to synchronize with.
[in]	done	Callback function to be executed.

Returns: Error status.

Implements nvdsinferserver::IPostprocessor.

Definition at line 209 of file sources/libs/nvdsinferserver/infer_iprocess.h.

◆ postHostProcess() [2/2]

NvDsInferStatus nvdsinferserver::BasePostprocessor::postHostProcess	(	SharedBatchArray	inBuf,
		SharedCuStream	mainStream,
		PostprocessDone	done
	)

inlineoverridevirtualinherited

Acquire an output buffer array and call host side processing steps.

Parameters

[in]	inBuf	Pointer to the input batch array.
[in]	mainStream	The main CUDA stream to synchronize with.
[in]	done	Callback function to be executed.

Returns: Error status.

Implements nvdsinferserver::IPostprocessor.

Definition at line 209 of file 9.1/sources/libs/nvdsinferserver/infer_iprocess.h.

◆ requestCudaOutBufs() [1/2]

SharedBatchArray nvdsinferserver::Postprocessor::requestCudaOutBufs ( const SharedBatchArray & inBuf )

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ requestCudaOutBufs() [2/2]

SharedBatchArray nvdsinferserver::Postprocessor::requestCudaOutBufs ( const SharedBatchArray & inBuf )

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ requestHostOutBufs() [1/2]

SharedBatchArray nvdsinferserver::Postprocessor::requestHostOutBufs ( const SharedBatchArray & inBuf )

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ requestHostOutBufs() [2/2]

SharedBatchArray nvdsinferserver::Postprocessor::requestHostOutBufs ( const SharedBatchArray & inBuf )

overrideprotectedvirtual

Implements nvdsinferserver::BasePostprocessor.

◆ setAllocator() [1/2]

void nvdsinferserver::Postprocessor::setAllocator	(	TensorAllocator	cpuAlloc,
		EventAllocator	event
	)

inline

Definition at line 82 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CpuAllocator, and m_EventAllocator.

◆ setAllocator() [2/2]

void nvdsinferserver::Postprocessor::setAllocator	(	TensorAllocator	cpuAlloc,
		EventAllocator	event
	)

inline

Definition at line 82 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CpuAllocator, and m_EventAllocator.

◆ setDllHandle() [1/2]

void nvdsinferserver::Postprocessor::setDllHandle ( const SharedDllHandle & dlHandle )

inline

Definition at line 70 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CustomLibHandle.

◆ setDllHandle() [2/2]

void nvdsinferserver::Postprocessor::setDllHandle ( const SharedDllHandle & dlHandle )

inline

Definition at line 70 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CustomLibHandle.

◆ setInputCopy() [1/2]

void nvdsinferserver::Postprocessor::setInputCopy ( bool enable )

inline

Definition at line 78 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CopyInputToHostBuffers.

◆ setInputCopy() [2/2]

void nvdsinferserver::Postprocessor::setInputCopy ( bool enable )

inline

Definition at line 78 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_CopyInputToHostBuffers.

◆ setLabelPath() [1/2]

void nvdsinferserver::Postprocessor::setLabelPath ( const std::string & path )

inline

Definition at line 73 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_LabelPath.

◆ setLabelPath() [2/2]

void nvdsinferserver::Postprocessor::setLabelPath ( const std::string & path )

inline

Definition at line 73 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_LabelPath.

◆ setNetworkInfo() [1/2]

void nvdsinferserver::Postprocessor::setNetworkInfo ( const NvDsInferNetworkInfo & info )

inline

Definition at line 74 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_NetworkInfo.

◆ setNetworkInfo() [2/2]

void nvdsinferserver::Postprocessor::setNetworkInfo ( const NvDsInferNetworkInfo & info )

inline

Definition at line 74 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_NetworkInfo.

◆ setOutputLayerCount() [1/2]

void nvdsinferserver::Postprocessor::setOutputLayerCount ( uint32_t num )

inline

Definition at line 77 of file sources/libs/nvdsinferserver/infer_postprocess.h.

References m_OutputLayerCount.

◆ setOutputLayerCount() [2/2]

void nvdsinferserver::Postprocessor::setOutputLayerCount ( uint32_t num )

inline

Definition at line 77 of file 9.1/sources/libs/nvdsinferserver/infer_postprocess.h.

References m_OutputLayerCount.

◆ setUniqueId() [1/2]

void nvdsinferserver::BasePostprocessor::setUniqueId ( int id )

inlineinherited

Definition at line 183 of file 9.1/sources/libs/nvdsinferserver/infer_iprocess.h.

◆ setUniqueId() [2/2]

void nvdsinferserver::BasePostprocessor::setUniqueId ( int id )

inlineinherited

Definition at line 183 of file sources/libs/nvdsinferserver/infer_iprocess.h.

◆ uniqueId() [1/2]

int nvdsinferserver::BasePostprocessor::uniqueId ( ) const

inlineinherited

Definition at line 184 of file sources/libs/nvdsinferserver/infer_iprocess.h.

◆ uniqueId() [2/2]

int nvdsinferserver::BasePostprocessor::uniqueId ( ) const

inlineinherited

Definition at line 184 of file 9.1/sources/libs/nvdsinferserver/infer_iprocess.h.

Field Documentation

◆ m_CopyInputToHostBuffers

bool nvdsinferserver::Postprocessor::m_CopyInputToHostBuffers = false

protected

Definition at line 117 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by needInputCopy(), and setInputCopy().

◆ m_CpuAllocator

TensorAllocator nvdsinferserver::Postprocessor::m_CpuAllocator

protected

Definition at line 126 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by setAllocator().

◆ m_CustomLibHandle

SharedDllHandle nvdsinferserver::Postprocessor::m_CustomLibHandle

protected

Custom library implementation.

Definition at line 116 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by setDllHandle().

◆ m_EventAllocator

EventAllocator nvdsinferserver::Postprocessor::m_EventAllocator

protected

Definition at line 127 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by setAllocator().

◆ m_LabelPath

std::string nvdsinferserver::Postprocessor::m_LabelPath

protected

Definition at line 121 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by setLabelPath().

◆ m_Labels

std::vector< std::vector< std::string > > nvdsinferserver::Postprocessor::m_Labels

protected

Holds the string labels for classes.

Definition at line 124 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by getLabels().

◆ m_NetworkInfo

NvDsInferNetworkInfo nvdsinferserver::Postprocessor::m_NetworkInfo = {0}

protected

Network input information.

Definition at line 119 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by setNetworkInfo().

◆ m_OutputLayerCount

uint32_t nvdsinferserver::Postprocessor::m_OutputLayerCount = 0

protected

Definition at line 120 of file sources/libs/nvdsinferserver/infer_postprocess.h.

Referenced by setOutputLayerCount().

The documentation for this class was generated from the following file:

sources/libs/nvdsinferserver/infer_postprocess.h


NVIDIA DeepStream SDK API Reference	9.1 Release

NVIDIA DeepStream SDK API Reference

9.1 Release

Detailed Description

Public Types

Public Member Functions

Protected Member Functions

Protected Attributes

Member Typedef Documentation

◆ EventAllocator [1/2]

◆ EventAllocator [2/2]

◆ PostprocessDone [1/2]

◆ PostprocessDone [2/2]

◆ TensorAllocator [1/2]

◆ TensorAllocator [2/2]

Constructor & Destructor Documentation

◆ Postprocessor() [1/2]

◆ ~Postprocessor() [1/2]

◆ Postprocessor() [2/2]

◆ ~Postprocessor() [2/2]

Member Function Documentation

◆ allocateResource() [1/2]

◆ allocateResource() [2/2]

◆ getLabels() [1/2]

◆ getLabels() [2/2]

◆ needInputCopy() [1/2]

◆ needInputCopy() [2/2]

◆ networkType() [1/2]

◆ networkType() [2/2]

◆ parseLabelsFile() [1/2]

◆ parseLabelsFile() [2/2]

◆ postCudaImpl() [1/2]

◆ postCudaImpl() [2/2]

◆ postCudaProcess() [1/2]

◆ postCudaProcess() [2/2]

◆ postHostImpl() [1/2]

◆ postHostImpl() [2/2]

◆ postHostProcess() [1/2]

◆ postHostProcess() [2/2]

◆ requestCudaOutBufs() [1/2]

◆ requestCudaOutBufs() [2/2]

◆ requestHostOutBufs() [1/2]

◆ requestHostOutBufs() [2/2]

◆ setAllocator() [1/2]

◆ setAllocator() [2/2]

◆ setDllHandle() [1/2]

◆ setDllHandle() [2/2]

◆ setInputCopy() [1/2]

◆ setInputCopy() [2/2]

◆ setLabelPath() [1/2]

◆ setLabelPath() [2/2]

◆ setNetworkInfo() [1/2]

◆ setNetworkInfo() [2/2]

◆ setOutputLayerCount() [1/2]

◆ setOutputLayerCount() [2/2]

◆ setUniqueId() [1/2]

◆ setUniqueId() [2/2]

◆ uniqueId() [1/2]

◆ uniqueId() [2/2]

Field Documentation

◆ m_CopyInputToHostBuffers

◆ m_CpuAllocator

◆ m_CustomLibHandle

◆ m_EventAllocator

◆ m_LabelPath

◆ m_Labels

◆ m_NetworkInfo

◆ m_OutputLayerCount