From eedc75400222ca1f8b030a221d8568ee26cb06d6 Mon Sep 17 00:00:00 2001 From: Will Badart Date: Fri, 16 Nov 2018 10:57:41 -0500 Subject: [PATCH] Fix getPoseMaxPeaks bug in Python API (#930) --- python/openpose/_openpose.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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() });