Fix getPoseMaxPeaks bug in Python API (#930)

eedc7540 · Will Badart · Gines · 0fff5a69 · eedc7540
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/openpose/_openpose.cpp python/openpose/_openpose.cpp +1 -1

未找到文件。
--- 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() });