keypointScaler.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_CORE_KEYPOINT_SCALER_HPP
#define OPENPOSE_CORE_KEYPOINT_SCALER_HPP
 
#include <openpose/core/common.hpp>
#include <openpose/core/enumClasses.hpp>
 
namespace op
{
    class OP_API KeypointScaler
    {
    public:
        explicit KeypointScaler(const ScaleMode scaleMode);
 
        virtual ~KeypointScaler();
 
        void scale(Array<float>& arrayToScale, const double scaleInputToOutput, const double scaleNetToOutput,
                   const Point<int>& producerSize) const;
 
        void scale(std::vector<Array<float>>& arraysToScale, const double scaleInputToOutput,
                   const double scaleNetToOutput, const Point<int>& producerSize) const;
 
        void scale(std::vector<std::vector<std::array<float,3>>>& poseCandidates, const double scaleInputToOutput,
                   const double scaleNetToOutput, const Point<int>& producerSize) const;
 
    private:
        const ScaleMode mScaleMode;
    };
}
 
#endif // OPENPOSE_CORE_KEYPOINT_SCALER_HPP