handExtractorNet.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_HAND_HAND_EXTRACTOR_HPP
#define OPENPOSE_HAND_HAND_EXTRACTOR_HPP
 
#include <atomic>
#include <openpose/core/common.hpp>
#include <openpose/core/enumClasses.hpp>
 
namespace op
{
    class OP_API HandExtractorNet
    {
    public:
        explicit HandExtractorNet(const Point<int>& netInputSize, const Point<int>& netOutputSize,
                                  const int numberScales = 1, const float rangeScales = 0.4f,
                                  const std::vector<HeatMapType>& heatMapTypes = {},
                                  const ScaleMode heatMapScaleMode = ScaleMode::ZeroToOneFixedAspect);
 
        virtual ~HandExtractorNet();
 
        void initializationOnThread();
 
        virtual void forwardPass(const std::vector<std::array<Rectangle<float>, 2>> handRectangles,
                                 const Matrix& cvInputData) = 0;
 
        std::array<Array<float>, 2> getHeatMaps() const;
 
        std::array<Array<float>, 2> getHandKeypoints() const;
 
        bool getEnabled() const;
 
        void setEnabled(const bool enabled);
 
    protected:
        const std::pair<int, float> mMultiScaleNumberAndRange;
        const Point<int> mNetOutputSize;
        Array<float> mHandImageCrop;
        std::array<Array<float>, 2> mHandKeypoints;
        // HeatMaps parameters
        const ScaleMode mHeatMapScaleMode;
        const std::vector<HeatMapType> mHeatMapTypes;
        std::array<Array<float>, 2> mHeatMaps;
        // Temporarily disable it
        std::atomic<bool> mEnabled;
 
        virtual void netInitializationOnThread() = 0;
 
    private:
        // Init with thread
        std::thread::id mThreadId;
 
        void checkThread() const;
 
        DELETE_COPY(HandExtractorNet);
    };
}
 
#endif // OPENPOSE_HAND_HAND_EXTRACTOR_HPP