OpenPose  1.0.0rc2
OpenPose: A Real-Time Multi-Person Key-Point Detection And Multi-Threading C++ Library
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
op::HandExtractorCaffe Class Reference

#include <handExtractorCaffe.hpp>

Inherits op::HandExtractor.

Public Member Functions

 HandExtractorCaffe (const Point< int > &netInputSize, const Point< int > &netOutputSize, const std::string &modelFolder, const int gpuId, const unsigned short numberScales=1, const float rangeScales=0.4f, const std::vector< HeatMapType > &heatMapTypes={}, const ScaleMode heatMapScale=ScaleMode::ZeroToOne, const bool enableGoogleLogging=true)
 
virtual ~HandExtractorCaffe ()
 
void netInitializationOnThread ()
 
void forwardPass (const std::vector< std::array< Rectangle< float >, 2 >> handRectangles, const cv::Mat &cvInputData, const double scaleInputToOutput)
 
- Public Member Functions inherited from op::HandExtractor
 HandExtractor (const Point< int > &netInputSize, const Point< int > &netOutputSize, const unsigned short numberScales=1, const float rangeScales=0.4f, const std::vector< HeatMapType > &heatMapTypes={}, const ScaleMode heatMapScale=ScaleMode::ZeroToOne)
 
virtual ~HandExtractor ()
 
void initializationOnThread ()
 
std::array< Array< float >, 2 > getHeatMaps () const
 
std::array< Array< float >, 2 > getHandKeypoints () const
 

Additional Inherited Members

- Protected Attributes inherited from op::HandExtractor
const std::pair< unsigned
short, float > 
mMultiScaleNumberAndRange
 
const Point< int > mNetOutputSize
 
Array< float > mHandImageCrop
 
std::array< Array< float >, 2 > mHandKeypoints
 
const ScaleMode mHeatMapScaleMode
 
const std::vector< HeatMapTypemHeatMapTypes
 
std::array< Array< float >, 2 > mHeatMaps
 

Detailed Description

Hand keypoint extractor class for Caffe framework.

Constructor & Destructor Documentation

op::HandExtractorCaffe::HandExtractorCaffe ( const Point< int > &  netInputSize,
const Point< int > &  netOutputSize,
const std::string &  modelFolder,
const int  gpuId,
const unsigned short  numberScales = 1,
const float  rangeScales = 0.4f,
const std::vector< HeatMapType > &  heatMapTypes = {},
const ScaleMode  heatMapScale = ScaleMode::ZeroToOne,
const bool  enableGoogleLogging = true 
)

Constructor of the HandExtractorCaffe class.

Parameters
netInputSizeSize at which the cropped image (where the hand is located) is resized.
netOutputSizeSize of the final results. At the moment, it must be equal than netOutputSize.
modelFolderFolder where the models are located.
gpuIdThe GPU index (0-based) which the deep net will use.
numberScalesNumber of scales to run. The more scales, the slower it will be but possibly also more accurate.
rangeScalesThe range between the smaller and bigger scale.
virtual op::HandExtractorCaffe::~HandExtractorCaffe ( )
virtual

Virtual destructor of the HandExtractor class. Required to allow inheritance.

Member Function Documentation

void op::HandExtractorCaffe::forwardPass ( const std::vector< std::array< Rectangle< float >, 2 >>  handRectangles,
const cv::Mat &  cvInputData,
const double  scaleInputToOutput 
)
virtual

This function extracts the hand keypoints for each detected hand in the image.

Parameters
handRectangleslocation of the hands in the image. It is a length-variable std::vector, where each index corresponds to a different person in the image. Internally the std::vector, a std::array of 2 elements: index 0 and 1 for left and right hand respectively. Inside each array element, a op::Rectangle<float> (similar to cv::Rect for floating values) with the position of that hand (or 0,0,0,0 if some hand is missing, e.g. if a specific person has only half of the body inside the image).
cvInputDataOriginal image in cv::Mat format and BGR format.
scaleInputToOutputDesired scale of the final keypoints. Set to 1 if the desired size is the cvInputData size.

Implements op::HandExtractor.

void op::HandExtractorCaffe::netInitializationOnThread ( )
virtual

This function must be call before using any other function. It must also be called inside the thread in which the functions are going to be used.

Implements op::HandExtractor.


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