| namespace op | |
| { | |
| /** | |
| * Hand keypoint extractor class for Caffe framework. | |
| */ | |
| class OP_API HandExtractorCaffe : public HandExtractorNet | |
| { | |
| public: | |
| /** | |
| * Constructor of the HandExtractorCaffe class. | |
| * @param netInputSize Size at which the cropped image (where the hand is located) is resized. | |
| * @param netOutputSize Size of the final results. At the moment, it must be equal than netOutputSize. | |
| * @param modelFolder Folder where the models are located. | |
| * @param gpuId The GPU index (0-based) which the deep net will use. | |
| * @param numberScales Number of scales to run. The more scales, the slower it will be but possibly also more | |
| * accurate. | |
| * @param rangeScales The range between the smaller and bigger scale. | |
| */ | |
| HandExtractorCaffe(const Point<int>& netInputSize, const Point<int>& netOutputSize, | |
| const std::string& modelFolder, const int gpuId, | |
| const int numberScales = 1, const float rangeScales = 0.4f, | |
| const std::vector<HeatMapType>& heatMapTypes = {}, | |
| const ScaleMode heatMapScaleMode = ScaleMode::ZeroToOneFixedAspect, | |
| const bool enableGoogleLogging = true); | |
| /** | |
| * Virtual destructor of the HandExtractor class. | |
| * Required to allow inheritance. | |
| */ | |
| virtual ~HandExtractorCaffe(); | |
| /** | |
| * 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. | |
| */ | |
| void netInitializationOnThread(); | |
| /** | |
| * This function extracts the hand keypoints for each detected hand in the image. | |
| * @param handRectangles location 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). | |
| * @param inputData Original image in Mat format and BGR format. | |
| */ | |
| void forwardPass(const std::vector<std::array<Rectangle<float>, 2>> handRectangles, const Matrix& inputData); | |
| private: | |
| // PIMPL idiom | |
| // http://www.cppsamples.com/common-tasks/pimpl.html | |
| struct ImplHandExtractorCaffe; | |
| std::unique_ptr<ImplHandExtractorCaffe> upImpl; | |
| Array<float> getHeatMapsFromLastPass() const; | |
| // PIMP requires DELETE_COPY & destructor, or extra code | |
| // http://oliora.github.io/2015/12/29/pimpl-and-rule-of-zero.html | |
| DELETE_COPY(HandExtractorCaffe); | |
| }; | |
| } | |