fixed poseNet.ObjectPose.FindKeypoint() Python function

shoaib42 · Feb 14, 2022 · b1afeb8 · b1afeb8
1 parent 91f0a70
commit b1afeb8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/docs/posenet.md b/docs/posenet.md
@@ -102,8 +102,8 @@ for pose in poses:
     # find the keypoint index from the list of detected keypoints
     # you can find these keypoint names in the model's JSON file, 
     # or with net.GetKeypointName() / net.GetNumKeypoints()
-    left_wrist_idx = net.FindKeypoint('left_wrist')
-    left_shoulder_idx = net.FindKeypoint('left_shoulder')
+    left_wrist_idx = pose.FindKeypoint('left_wrist')
+    left_shoulder_idx = pose.FindKeypoint('left_shoulder')
 
     # if the keypoint index is < 0, it means it wasn't found in the image
     if left_wrist_idx < 0 or left_shoulder_idx < 0:

diff --git a/python/bindings/PyPoseNet.cpp b/python/bindings/PyPoseNet.cpp
@@ -670,11 +670,14 @@ static PyObject* PyObjectPose_FindKeypoint( PyObjectPose_Object* self, PyObject
 		return NULL;
 	}
 
-	// parse arguments
+	// this function accepts either the keypoint ID (int) or the name (string)
+	// in the string case, it will just look up the keypoint ID for you
 	int id = 0;
 
 	if( !PyArg_ParseTuple(args, "i", &id) )
 	{
+		PyErr_Clear();  // PyArg_ParseTuple will throw an exception
+
 		if( self->net != NULL )
 		{
 			const char* name = NULL;