faceExtractorCaffe.hpp

Go to the documentation of this file.

#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
{
    class OP_API FaceExtractorCaffe : public FaceExtractorNet
    {
    public:
        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();
 
        void netInitializationOnThread();
 
        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