camenduru's picture
thanks to openpose ❤
7fc5a59
#ifndef OPENPOSE_CORE_CV_MAT_TO_OP_OUTPUT_HPP
#define OPENPOSE_CORE_CV_MAT_TO_OP_OUTPUT_HPP
#include <openpose/core/common.hpp>
namespace op
{
class OP_API CvMatToOpOutput
{
public:
CvMatToOpOutput(const bool gpuResize = false);
virtual ~CvMatToOpOutput();
std::tuple<std::shared_ptr<float*>, std::shared_ptr<bool>, std::shared_ptr<unsigned long long>>
getSharedParameters();
Array<float> createArray(
const Matrix& inputData, const double scaleInputToOutput, const Point<int>& outputResolution);
private:
const bool mGpuResize;
unsigned char* pInputImageCuda;
std::shared_ptr<float*> spOutputImageCuda;
unsigned long long pInputMaxSize;
std::shared_ptr<unsigned long long> spOutputMaxSize;
std::shared_ptr<bool> spGpuMemoryAllocated;
};
}
#endif // OPENPOSE_CORE_CV_MAT_TO_OP_OUTPUT_HPP