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