| | #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 |
| |
|