1 #ifndef OPENPOSE_POSE_BODY_PART_CONNECTOR_CAFFE_HPP
2 #define OPENPOSE_POSE_BODY_PART_CONNECTOR_CAFFE_HPP
22 virtual inline const char*
type()
const {
return "BodyPartConnector"; }
53 T mInterMinAboveThreshold;
58 std::array<int, 4> mHeatMapsSize;
59 std::array<int, 4> mPeaksSize;
60 std::array<int, 4> mTopSize;
62 unsigned int* pBodyPartPairsGpuPtr;
63 unsigned int* pMapIdxGpuPtr;
65 T* pFinalOutputGpuPtr;
71 #endif // OPENPOSE_POSE_BODY_PART_CONNECTOR_CAFFE_HPP
void setMinSubsetCnt(const int minSubsetCnt)
virtual void Forward_gpu(const std::vector< caffe::Blob< T > * > &bottom, Array< T > &poseKeypoints, Array< T > &poseScores)
void setScaleNetToOutput(const T scaleNetToOutput)
PoseModel
Definition: enumClasses.hpp:9
virtual void Backward_cpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)
virtual void Forward_cpu(const std::vector< caffe::Blob< T > * > &bottom, Array< T > &poseKeypoints, Array< T > &poseScores)
virtual ~BodyPartConnectorCaffe()
virtual void Reshape(const std::vector< caffe::Blob< T > * > &bottom)
virtual const char * type() const
Definition: bodyPartConnectorCaffe.hpp:22
void setMinSubsetScore(const T minSubsetScore)
void setInterMinAboveThreshold(const T interMinAboveThreshold)
Definition: macros.hpp:81
void setPoseModel(const PoseModel poseModel)
std::vector< T, Alloc > vector
Definition: cl2.hpp:567
void setInterThreshold(const T interThreshold)
virtual void Forward(const std::vector< caffe::Blob< T > * > &bottom, Array< T > &poseKeypoints, Array< T > &poseScores)
Definition: bodyPartConnectorCaffe.hpp:13
virtual void Backward_gpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)