diff --git a/python/openpose/_openpose.cpp b/python/openpose/_openpose.cpp index ea7dd716..e20720b9 100644 --- a/python/openpose/_openpose.cpp +++ b/python/openpose/_openpose.cpp @@ -188,7 +188,7 @@ public: 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() });