Fixes iOS hand landmarker connections

PiperOrigin-RevId: 565442497
google-ai-edge · Sep 14, 2023 · 85b1938 · 85b1938
1 parent 124a4de
commit 85b1938
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/mediapipe/tasks/ios/vision/hand_landmarker/sources/MPPHandLandmarker.h b/mediapipe/tasks/ios/vision/hand_landmarker/sources/MPPHandLandmarker.h
@@ -33,6 +33,8 @@ typedef NS_ENUM(NSUInteger, MPPHandLandmark) {
 
   MPPHandLandmarkThumbIP,
 
+  MPPHandLandmarkThumbTIP,
+
   MPPHandLandmarkIndexFingerMCP,
 
   MPPHandLandmarkIndexFingerPIP,
@@ -78,6 +80,9 @@ NS_SWIFT_NAME(HandLandmarker)
 /** The array of connections between the landmarks in the palm. */
 @property(class, nonatomic, readonly) NSArray<MPPConnection *> *handPalmConnections;
 
+/** The array of connections between the landmarks in the thumb. */
+@property(class, nonatomic, readonly) NSArray<MPPConnection *> *handThumbConnections;
+
 /** The array of connections between the landmarks in the index finger. */
 @property(class, nonatomic, readonly) NSArray<MPPConnection *> *handIndexFingerConnections;
 

diff --git a/mediapipe/tasks/ios/vision/hand_landmarker/sources/MPPHandLandmarker.mm b/mediapipe/tasks/ios/vision/hand_landmarker/sources/MPPHandLandmarker.mm
@@ -169,6 +169,10 @@ - (BOOL)detectAsyncInImage:(MPPImage *)image
   return MPPHandPalmConnections;
 }
 
++ (NSArray<MPPConnection *> *)handThumbConnections {
+  return MPPHandThumbConnections;
+}
+
 + (NSArray<MPPConnection *> *)handIndexFingerConnections {
   return MPPHandIndexFingerConnections;
 }

diff --git a/mediapipe/tasks/ios/vision/hand_landmarker/sources/MPPHandLandmarksConnections.h b/mediapipe/tasks/ios/vision/hand_landmarker/sources/MPPHandLandmarksConnections.h
@@ -44,8 +44,8 @@ NSArray<MPPConnection *> *const MPPHandRingFingerConnections = @[
 ];
 
 NSArray<MPPConnection *> *const MPPHandPinkyConnections = @[
-  [[MPPConnection alloc] initWithStart:16 end:17], [[MPPConnection alloc] initWithStart:17 end:18],
-  [[MPPConnection alloc] initWithStart:18 end:19]
+  [[MPPConnection alloc] initWithStart:17 end:18], [[MPPConnection alloc] initWithStart:18 end:19],
+  [[MPPConnection alloc] initWithStart:19 end:20]
 ];
 
 NSArray<MPPConnection *> *const MPPHandConnections = [[[[[[NSArray