Fix getPoseMaxPeaks bug in Python API (CMU-Perceptual-Computing-Lab#930)

aslily1234 · Nov 16, 2018 · eedc754 · eedc754
1 parent 0fff5a6
commit eedc754
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/openpose/_openpose.cpp b/python/openpose/_openpose.cpp
@@ -188,7 +188,7 @@ OP_API class OpenPose {
         resizeAndMergeCaffe->Reshape(caffeNetOutputBlobs, { heatMapsBlob.get() },
             op::getPoseNetDecreaseFactor(poseModel), 1.f / 1.f, true,
             0);
-        nmsCaffe->Reshape({ heatMapsBlob.get() }, { peaksBlob.get() }, op::getPoseMaxPeaks(poseModel),
+        nmsCaffe->Reshape({ heatMapsBlob.get() }, { peaksBlob.get() }, op::getPoseMaxPeaks(),
             op::getPoseNumberBodyParts(poseModel), 0);
         bodyPartConnectorCaffe->Reshape({ heatMapsBlob.get(), peaksBlob.get() });