1 #ifndef OPENPOSE_POSE_BODY_PARTS_CONNECTOR_HPP
2 #define OPENPOSE_POSE_BODY_PARTS_CONNECTOR_HPP
11 const T*
const peaksPtr,
const PoseModel poseModel,
const Point<int>& heatMapSize,
12 const int maxPeaks,
const T interMinAboveThreshold,
const T interThreshold,
13 const int minSubsetCnt,
const T minSubsetScore,
const T scaleFactor = 1.f);
17 const T*
const peaksPtr,
const PoseModel poseModel,
const Point<int>& heatMapSize,
18 const int maxPeaks,
const T interMinAboveThreshold,
const T interThreshold,
19 const int minSubsetCnt,
const T minSubsetScore,
const T scaleFactor = 1.f,
20 Array<T> pairScoresCpu = Array<T>{}, T* pairScoresGpuPtr =
nullptr,
21 const unsigned int*
const bodyPartPairsGpuPtr =
nullptr,
22 const unsigned int*
const mapIdxGpuPtr =
nullptr,
23 const T*
const peaksGpuPtr =
nullptr);
28 const T*
const heatMapPtr,
const T*
const peaksPtr,
const PoseModel poseModel,
const Point<int>& heatMapSize,
29 const int maxPeaks,
const T interThreshold,
const T interMinAboveThreshold,
30 const std::vector<unsigned int>& bodyPartPairs,
const unsigned int numberBodyParts,
31 const unsigned int numberBodyPartPairs,
const Array<T>& precomputedPAFs = Array<T>());
35 const std::vector<std::pair<std::vector<int>, T>>& subsets,
36 const unsigned int numberBodyParts,
const int minSubsetCnt,
37 const T minSubsetScore,
const int maxPeaks);
41 const std::vector<std::pair<std::vector<int>, T>>& subsets,
42 const std::vector<int>& validSubsetIndexes,
const T*
const peaksPtr,
43 const int numberPeople,
const unsigned int numberBodyParts,
44 const unsigned int numberBodyPartPairs);
48 const Array<T>& pairScores,
const T*
const peaksPtr,
const int maxPeaks,
49 const std::vector<unsigned int>& bodyPartPairs,
const unsigned int numberBodyPartPairs);
53 const std::vector<std::tuple<T, T, int, int, int>>& pairScores,
const T*
const peaksPtr,
const int maxPeaks,
54 const std::vector<unsigned int>& bodyPartPairs,
const unsigned int numberBodyParts);
57 #endif // OPENPOSE_POSE_BODY_PARTS_CONNECTOR_HPP
OP_API std::vector< std::pair< std::vector< int >, T > > createPeopleVector(const T *const heatMapPtr, const T *const peaksPtr, const PoseModel poseModel, const Point< int > &heatMapSize, const int maxPeaks, const T interThreshold, const T interMinAboveThreshold, const std::vector< unsigned int > &bodyPartPairs, const unsigned int numberBodyParts, const unsigned int numberBodyPartPairs, const Array< T > &precomputedPAFs=Array< T >())
PoseModel
Definition: enumClasses.hpp:9
OP_API void removePeopleBelowThresholds(std::vector< int > &validSubsetIndexes, int &numberPeople, const std::vector< std::pair< std::vector< int >, T >> &subsets, const unsigned int numberBodyParts, const int minSubsetCnt, const T minSubsetScore, const int maxPeaks)
OP_API void connectBodyPartsCpu(Array< T > &poseKeypoints, Array< T > &poseScores, const T *const heatMapPtr, const T *const peaksPtr, const PoseModel poseModel, const Point< int > &heatMapSize, const int maxPeaks, const T interMinAboveThreshold, const T interThreshold, const int minSubsetCnt, const T minSubsetScore, const T scaleFactor=1.f)
OP_API std::vector< std::tuple< T, T, int, int, int > > pafPtrIntoVector(const Array< T > &pairScores, const T *const peaksPtr, const int maxPeaks, const std::vector< unsigned int > &bodyPartPairs, const unsigned int numberBodyPartPairs)
OP_API void peopleVectorToPeopleArray(Array< T > &poseKeypoints, Array< T > &poseScores, const T scaleFactor, const std::vector< std::pair< std::vector< int >, T >> &subsets, const std::vector< int > &validSubsetIndexes, const T *const peaksPtr, const int numberPeople, const unsigned int numberBodyParts, const unsigned int numberBodyPartPairs)
OP_API std::vector< std::pair< std::vector< int >, T > > pafVectorIntoPeopleVector(const std::vector< std::tuple< T, T, int, int, int >> &pairScores, const T *const peaksPtr, const int maxPeaks, const std::vector< unsigned int > &bodyPartPairs, const unsigned int numberBodyParts)
std::vector< T, Alloc > vector
Definition: cl2.hpp:567
#define OP_API
Definition: macros.hpp:18
OP_API void connectBodyPartsGpu(Array< T > &poseKeypoints, Array< T > &poseScores, const T *const heatMapGpuPtr, const T *const peaksPtr, const PoseModel poseModel, const Point< int > &heatMapSize, const int maxPeaks, const T interMinAboveThreshold, const T interThreshold, const int minSubsetCnt, const T minSubsetScore, const T scaleFactor=1.f, Array< T > pairScoresCpu=Array< T >{}, T *pairScoresGpuPtr=nullptr, const unsigned int *const bodyPartPairsGpuPtr=nullptr, const unsigned int *const mapIdxGpuPtr=nullptr, const T *const peaksGpuPtr=nullptr)