File size: 903 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_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
|