File size: 987 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 | #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
|