faceExtractorNet.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_FACE_FACE_EXTRACTOR_HPP
#define OPENPOSE_FACE_FACE_EXTRACTOR_HPP
 
#include <atomic>
#include <openpose/core/common.hpp>
#include <openpose/core/enumClasses.hpp>
 
namespace op
{
    class OP_API FaceExtractorNet
    {
    public:
        explicit FaceExtractorNet(const Point<int>& netInputSize, const Point<int>& netOutputSize,
                                  const std::vector<HeatMapType>& heatMapTypes = {},
                                  const ScaleMode heatMapScaleMode = ScaleMode::ZeroToOneFixedAspect);
 
        virtual ~FaceExtractorNet();
 
        void initializationOnThread();
 
        virtual void forwardPass(const std::vector<Rectangle<float>>& faceRectangles, const Matrix& inputData) = 0;
 
        Array<float> getHeatMaps() const;
 
        Array<float> getFaceKeypoints() const;
 
        bool getEnabled() const;
 
        void setEnabled(const bool enabled);
 
    protected:
        const Point<int> mNetOutputSize;
        Array<float> mFaceImageCrop;
        Array<float> mFaceKeypoints;
        // HeatMaps parameters
        Array<float> mHeatMaps;
        const ScaleMode mHeatMapScaleMode;
        const std::vector<HeatMapType> mHeatMapTypes;
        // Temporarily disable it
        std::atomic<bool> mEnabled;
 
        virtual void netInitializationOnThread() = 0;
 
    private:
        // Init with thread
        std::thread::id mThreadId;
 
        void checkThread() const;
 
        DELETE_COPY(FaceExtractorNet);
    };
}
 
#endif // OPENPOSE_FACE_FACE_EXTRACTOR_HPP