File size: 4,018 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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | #ifndef OPENPOSE_POSE_BODY_PARTS_CONNECTOR_HPP
#define OPENPOSE_POSE_BODY_PARTS_CONNECTOR_HPP
#include <openpose/core/common.hpp>
#include <openpose/pose/enumClasses.hpp>
namespace op
{
template <typename T>
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 defaultNmsThreshold,
const T scaleFactor = 1.f, const bool maximizePositives = false);
// Windows: Cuda functions do not include OP_API
template <typename T>
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 defaultNmsThreshold,
const T scaleFactor, const bool maximizePositives, Array<T> pairScoresCpu, T* pairScoresGpuPtr,
const unsigned int* const bodyPartPairsGpuPtr, const unsigned int* const mapIdxGpuPtr,
const T* const peaksGpuPtr);
template <typename T>
void connectBodyPartsOcl(
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 defaultNmsThreshold,
const T scaleFactor = 1.f, const bool maximizePositives = false,
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, const int gpuID = 0);
// Private functions used by the 2 above functions
template <typename T>
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 T defaultNmsThreshold,
const Array<T>& precomputedPAFs = Array<T>());
template <typename T>
void removePeopleBelowThresholdsAndFillFaces(
std::vector<int>& validSubsetIndexes, int& numberPeople,
std::vector<std::pair<std::vector<int>, T>>& subsets, const unsigned int numberBodyParts,
const int minSubsetCnt, const T minSubsetScore, const bool maximizePositives, const T* const peaksPtr);
template <typename T>
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);
template <typename T>
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);
template <typename T>
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);
}
#endif // OPENPOSE_POSE_BODY_PARTS_CONNECTOR_HPP
|