Fix getPoseMaxPeaks bug in Python API (#930)

This commit is contained in:
Will Badart
2018-11-16 10:57:41 -05:00
committed by Gines
parent 0fff5a69b7
commit eedc754002
+1 -1
View File
@@ -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() });