Inheritance diagram for HpeAssociativeEmbedding:

Collaboration diagram for HpeAssociativeEmbedding:

[legend]

Public Member Functions
	HpeAssociativeEmbedding (const std::string &modelFileName, double aspectRatio, int targetSize, float confidenceThreshold, const std::string &layout="", float delta=0.0, RESIZE_MODE resizeMode=RESIZE_KEEP_ASPECT)

std::unique_ptr< ResultBase >	postprocess (InferenceResult &infResult) override

std::shared_ptr< InternalModelData >	preprocess (const InputData &inputData, ov::InferRequest &request) override

Public Member Functions inherited from ImageModel
	ImageModel (const std::string &modelFileName, bool useAutoResize, const std::string &layout="")

std::shared_ptr< InternalModelData >	preprocess (const InputData &inputData, ov::InferRequest &request) override

Public Member Functions inherited from ModelBase
	ModelBase (const std::string &modelFileName, const std::string &layout="")

virtual ov::CompiledModel	compileModel (const ModelConfig &config, ov::Core &core)

virtual void	onLoadCompleted (const std::vector< ov::InferRequest > &requests)

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

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

std::string	getModelFileName ()

void	setInputsPreprocessing (bool reverseInputChannels, const std::string &meanValues, const std::string &scaleValues)

Protected Member Functions
void	prepareInputsOutputs (std::shared_ptr< ov::Model > &model) override

void	changeInputSize (std::shared_ptr< ov::Model > &model)

std::string	findTensorByName (const std::string &tensorName, const std::vector< std::string > &outputsNames)

std::vector< cv::Mat >	split (float *data, const ov::Shape &shape)

std::vector< HumanPose >	extractPoses (std::vector< cv::Mat > &heatMaps, const std::vector< cv::Mat > &aembdsMaps, const std::vector< cv::Mat > &nmsHeatMaps) const

Protected Member Functions inherited from ModelBase
virtual void	setBatch (std::shared_ptr< ov::Model > &model)

std::shared_ptr< ov::Model >	prepareModel (ov::Core &core)

ov::Layout	getInputLayout (const ov::Output< ov::Node > &input)

Protected Attributes
cv::Size	inputLayerSize

double	aspectRatio

int	targetSize

float	confidenceThreshold

float	delta

std::string	embeddingsTensorName

std::string	heatmapsTensorName

std::string	nmsHeatmapsTensorName

Protected Attributes inherited from ImageModel
bool	useAutoResize

size_t	netInputHeight = 0

size_t	netInputWidth = 0

cv::InterpolationFlags	interpolationMode = cv::INTER_LINEAR

RESIZE_MODE	resizeMode = RESIZE_FILL

Protected Attributes inherited from ModelBase
InputTransform	inputTransform = InputTransform()

std::vector< std::string >	inputsNames

std::vector< std::string >	outputsNames

ov::CompiledModel	compiledModel

std::string	modelFileName

ModelConfig	config = {}

std::map< std::string, ov::Layout >	inputsLayouts

Static Protected Attributes
static const int	numJoints = 17

static const int	stride = 32

static const int	maxNumPeople = 30

static const cv::Vec3f	meanPixel = cv::Vec3f::all(128)

static const float	detectionThreshold = 0.1f

static const float	tagThreshold = 1.0f

Constructor & Destructor Documentation

◆ HpeAssociativeEmbedding()

HpeAssociativeEmbedding::HpeAssociativeEmbedding	(	const std::string &	modelFileName,
		double	aspectRatio,
		int	targetSize,
		float	confidenceThreshold,
		const std::string &	layout = `""`,
		float	delta = `0.0`,
		RESIZE_MODE	resizeMode = `RESIZE_KEEP_ASPECT`
	)

Constructor

Parameters

modelFileName	name of model to load
aspectRatio	- the ratio of input width to its height.
targetSize	- the length of a short image side used for model reshaping.
confidenceThreshold	- threshold to eliminate low-confidence poses. Any pose with confidence lower than this threshold will be ignored.
layout	- model input layout

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