usr/local/include/openpose/core/cvMatToOpInput.hpp

#ifndef OPENPOSE_CORE_CV_MAT_TO_OP_INPUT_HPP
#define OPENPOSE_CORE_CV_MAT_TO_OP_INPUT_HPP

#include <openpose/core/common.hpp>
#include <openpose/pose/enumClasses.hpp>

namespace op
{
    class OP_API CvMatToOpInput
    {
    public:
        CvMatToOpInput(const PoseModel poseModel = PoseModel::BODY_25, const bool gpuResize = false);

        virtual ~CvMatToOpInput();

        std::vector<Array<float>> createArray(
            const Matrix& inputData, const std::vector<double>& scaleInputToNetInputs,
            const std::vector<Point<int>>& netInputSizes);

    private:
        const PoseModel mPoseModel;
        const bool mGpuResize;
        unsigned char* pInputImageCuda;
        float* pInputImageReorderedCuda;
        float* pOutputImageCuda;
        unsigned long long pInputMaxSize;
        unsigned long long pOutputMaxSize;
    };
}

#endif // OPENPOSE_CORE_CV_MAT_TO_OP_INPUT_HPP