cvcore::bodypose2d::BodyPose2DPostProcessor Class Reference

Detailed Description

Interface for running post-processing on bodypose2d network.

Definition at line 224 of file BodyPose2D.h.

Public Member Functions
	BodyPose2DPostProcessor ()=delete
	Default constructor is deleted. More...
	BodyPose2DPostProcessor (const BodyPose2DPostProcessorParams &params, const ModelInputParams &modelInputParams)
	Constructor of BodyPose2DPostProcessor. More...
	~BodyPose2DPostProcessor ()
	Destructor of BodyPose2DPostProcessor. More...
void	allocateStagingBuffers ()
	Allocate staging CPU buffers (used when inputs are GPU Tensors). More...
void	execute (Array< ArrayN< Human, BodyPose2D::MAX_HUMAN_COUNT >> &output, const Tensor< NCHW, CX, F32 > &pafMap, const Tensor< NCHW, CX, F32 > &heatMap, int imageWidth, int imageHeight, cudaStream_t stream=0)
	Main interface to run post-processing for batch input. More...

Constructor & Destructor Documentation

BodyPose2DPostProcessor() [1/2]

cvcore::bodypose2d::BodyPose2DPostProcessor::BodyPose2DPostProcessor ( )

delete

Default constructor is deleted.

BodyPose2DPostProcessor() [2/2]

cvcore::bodypose2d::BodyPose2DPostProcessor::BodyPose2DPostProcessor	(	const BodyPose2DPostProcessorParams &	params,
		const ModelInputParams &	modelInputParams
	)

Constructor of BodyPose2DPostProcessor.

Parameters

params	bodypose2d post-processing parameters.
modelInputParams	model paramters for network.

~BodyPose2DPostProcessor()

cvcore::bodypose2d::BodyPose2DPostProcessor::~BodyPose2DPostProcessor

(

)

Destructor of BodyPose2DPostProcessor.

Member Function Documentation

allocateStagingBuffers()

void cvcore::bodypose2d::BodyPose2DPostProcessor::allocateStagingBuffers

(

)

Allocate staging CPU buffers (used when inputs are GPU Tensors).

execute()

void cvcore::bodypose2d::BodyPose2DPostProcessor::execute	(	Array< ArrayN< Human, BodyPose2D::MAX_HUMAN_COUNT >> &	output,
		const Tensor< NCHW, CX, F32 > &	pafMap,
		const Tensor< NCHW, CX, F32 > &	heatMap,
		int	imageWidth,
		int	imageHeight,
		cudaStream_t	stream = 0
	)

Main interface to run post-processing for batch input.

Parameters

output	output bodypose2d detection results.
pafMap	input part-affinity-field layer.
heatMap	input heatmap layer.
imageWidth	input image width.
imageHeight	input image height.

---

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

• BodyPose2D.h