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