diff --git a/include/openpose/pose/bodyPartConnectorBase.hpp b/include/openpose/pose/bodyPartConnectorBase.hpp index 840de2e264ac76b2eae70ddbae78e6670fb9ad86..62033e7c0383b214c9b0b122d5d4144ef660f631 100644 --- a/include/openpose/pose/bodyPartConnectorBase.hpp +++ b/include/openpose/pose/bodyPartConnectorBase.hpp @@ -31,7 +31,7 @@ namespace op OP_API void removeSubsetsBelowThresholds(std::vector& validSubsetIndexes, int& numberPeople, const std::vector, double>>& subsets, const unsigned int subsetCounterIndex, const unsigned int numberBodyParts, - const int minSubsetCnt, const T minSubsetScore); + const int minSubsetCnt, const T minSubsetScore, const int maxPeaks); template OP_API void subsetsToPoseKeypointsAndScores(Array& poseKeypoints, Array& poseScores, const T scaleFactor, diff --git a/src/openpose/pose/bodyPartConnectorBase.cu b/src/openpose/pose/bodyPartConnectorBase.cu index 666171ed4973d927ba9d288786044fbcd221c6a2..f54eaf5d4eae388bdce171d9a29531f190f23e98 100644 --- a/src/openpose/pose/bodyPartConnectorBase.cu +++ b/src/openpose/pose/bodyPartConnectorBase.cu @@ -38,7 +38,7 @@ namespace op std::vector validSubsetIndexes; validSubsetIndexes.reserve(fastMin((size_t)POSE_MAX_PEOPLE, subsets.size())); removeSubsetsBelowThresholds(validSubsetIndexes, numberPeople, subsets, subsetCounterIndex, - numberBodyParts, minSubsetCnt, minSubsetScore); + numberBodyParts, minSubsetCnt, minSubsetScore, maxPeaks); // Fill and return poseKeypoints subsetsToPoseKeypointsAndScores(poseKeypoints, poseScores, scaleFactor, subsets, validSubsetIndexes,