File size: 2,508 Bytes
7fc5a59 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | #ifndef OPENPOSE_FACE_FACE_EXTRACTOR_CAFFE_HPP
#define OPENPOSE_FACE_FACE_EXTRACTOR_CAFFE_HPP
#include <openpose/core/common.hpp>
#include <openpose/core/enumClasses.hpp>
#include <openpose/face/faceExtractorNet.hpp>
namespace op
{
/**
* Face keypoint extractor class for Caffe framework.
*/
class OP_API FaceExtractorCaffe : public FaceExtractorNet
{
public:
/**
* Constructor of the FaceExtractor class.
* @param netInputSize Size at which the cropped image (where the face is located) is resized.
* @param netOutputSize Size of the final results. At the moment, it must be equal than netOutputSize.
*/
FaceExtractorCaffe(const Point<int>& netInputSize, const Point<int>& netOutputSize,
const std::string& modelFolder, const int gpuId,
const std::vector<HeatMapType>& heatMapTypes = {},
const ScaleMode heatMapScaleMode = ScaleMode::ZeroToOneFixedAspect,
const bool enableGoogleLogging = true);
virtual ~FaceExtractorCaffe();
/**
* 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 face keypoints for each detected face in the image.
* @param faceRectangles location of the faces in the image. It is a length-variable std::vector, where
* each index corresponds to a different person in the image. Internally, a op::Rectangle<float>
* (similar to cv::Rect for floating values) with the position of that face (or 0,0,0,0 if
* some face is missing, e.g., if a specific person has only half of the body inside the image).
* @param cvInputData Original image in Mat format and BGR format.
*/
void forwardPass(const std::vector<Rectangle<float>>& faceRectangles, const Matrix& inputData);
private:
// PIMPL idiom
// http://www.cppsamples.com/common-tasks/pimpl.html
struct ImplFaceExtractorCaffe;
std::unique_ptr<ImplFaceExtractorCaffe> upImpl;
// PIMP requires DELETE_COPY & destructor, or extra code
// http://oliora.github.io/2015/12/29/pimpl-and-rule-of-zero.html
DELETE_COPY(FaceExtractorCaffe);
};
}
#endif // OPENPOSE_FACE_FACE_EXTRACTOR_CAFFE_HPP
|