#include <poseExtractorCaffe.hpp>

Inheritance diagram for op::PoseExtractorCaffe:

Public Member Functions
	PoseExtractorCaffe (const PoseModel poseModel, const std::string &modelFolder, const int gpuId, const std::vector< HeatMapType > &heatMapTypes={}, const ScaleMode heatMapScaleMode=ScaleMode::ZeroToOneFixedAspect, const bool addPartCandidates=false, const bool maximizePositives=false, const std::string &protoTxtPath="", const std::string &caffeModelPath="", const float upsamplingRatio=0.f, const bool enableNet=true, const bool enableGoogleLogging=true)

virtual	~PoseExtractorCaffe ()

virtual void	netInitializationOnThread ()

virtual void	forwardPass (const std::vector< Array< float >> &inputNetData, const Point< int > &inputDataSize, const std::vector< double > &scaleInputToNetInputs={1.f}, const Array< float > &poseNetOutput=Array< float >{})

const float *	getCandidatesCpuConstPtr () const

const float *	getCandidatesGpuConstPtr () const

const float *	getHeatMapCpuConstPtr () const

const float *	getHeatMapGpuConstPtr () const

std::vector< int >	getHeatMapSize () const

const float *	getPoseGpuConstPtr () const

Public Member Functions inherited from op::PoseExtractorNet
	PoseExtractorNet (const PoseModel poseModel, const std::vector< HeatMapType > &heatMapTypes={}, const ScaleMode heatMapScaleMode=ScaleMode::ZeroToOneFixedAspect, const bool addPartCandidates=false, const bool maximizePositives=false)

virtual	~PoseExtractorNet ()

void	initializationOnThread ()

Array< float >	getHeatMapsCopy () const

std::vector< std::vector< std::array< float, 3 > > >	getCandidatesCopy () const

Array< float >	getPoseKeypoints () const

Array< float >	getPoseScores () const

float	getScaleNetToOutput () const

double	get (const PoseProperty property) const

void	set (const PoseProperty property, const double value)

void	increase (const PoseProperty property, const double value)

void	clear ()

Additional Inherited Members
Protected Member Functions inherited from op::PoseExtractorNet
void	checkThread () const

Protected Attributes inherited from op::PoseExtractorNet
const PoseModel	mPoseModel

Point< int >	mNetOutputSize

Array< float >	mPoseKeypoints

Array< float >	mPoseScores

float	mScaleNetToOutput

Detailed Description

Definition at line 16 of file poseExtractorCaffe.hpp.

Constructor & Destructor Documentation

◆ PoseExtractorCaffe()

op::PoseExtractorCaffe::PoseExtractorCaffe	(	const PoseModel	poseModel,
		const std::string &	modelFolder,
		const int	gpuId,
		const std::vector< HeatMapType > &	heatMapTypes = `{}`,
		const ScaleMode	heatMapScaleMode = `ScaleMode::ZeroToOneFixedAspect`,
		const bool	addPartCandidates = `false`,
		const bool	maximizePositives = `false`,
		const std::string &	protoTxtPath = `""`,
		const std::string &	caffeModelPath = `""`,
		const float	upsamplingRatio = `0.f`,
		const bool	enableNet = `true`,
		const bool	enableGoogleLogging = `true`
	)

◆ ~PoseExtractorCaffe()

virtual op::PoseExtractorCaffe::~PoseExtractorCaffe ( )

virtual

Member Function Documentation

◆ forwardPass()

virtual void op::PoseExtractorCaffe::forwardPass	(	const std::vector< Array< float >> &	inputNetData,
		const Point< int > &	inputDataSize,
		const std::vector< double > &	scaleInputToNetInputs = `{1.f}`,
		const Array< float > &	poseNetOutput = `Array< float >{}`
	)

virtual

Parameters

poseNetOutput If it is not empty, OpenPose will not run its internal body pose estimation network and will instead use this data as the substitute of its network. The size of this element must match the size of the output of its internal network, or it will lead to core dumped (segmentation) errors. You can modify the pose estimation flags to match the dimension of both elements (e.g., --net_resolution, --scale_number, etc.).

Implements op::PoseExtractorNet.