cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added optional removal of Z position in gesture recognizer test util

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-05-25 19:13:22 +05:30
parent 4438606f3f
commit f51dd05e89
3 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mediapipe/tasks/ios/test/vision/gesture_recognizer/MPPGestureRecognizerTests.m
View File

@ -98,7 +98,8 @@ static const float kLandmarksErrorTolerance = 0.03f;
return [MPPGestureRecognizerResult return [MPPGestureRecognizerResult
gestureRecognizerResultsFromTextEncodedProtobufFileWithName:filePath gestureRecognizerResultsFromTextEncodedProtobufFileWithName:filePath
gestureLabel:kExpectedThumbUpLabel]; gestureLabel:kExpectedThumbUpLabel
shouldRemoveZPosition:YES];
} }
+ (MPPGestureRecognizerResult *)fistGestureRecognizerResultWithLabel:(NSString *)gestureLabel { + (MPPGestureRecognizerResult *)fistGestureRecognizerResultWithLabel:(NSString *)gestureLabel {
@ -106,7 +107,8 @@ static const float kLandmarksErrorTolerance = 0.03f;
return [MPPGestureRecognizerResult return [MPPGestureRecognizerResult
gestureRecognizerResultsFromTextEncodedProtobufFileWithName:filePath gestureRecognizerResultsFromTextEncodedProtobufFileWithName:filePath
gestureLabel:gestureLabel]; gestureLabel:gestureLabel
shouldRemoveZPosition:YES];
} }
- (void)assertMultiHandLandmarks:(NSArray<NSArray<MPPNormalizedLandmark *> *> *)multiHandLandmarks - (void)assertMultiHandLandmarks:(NSArray<NSArray<MPPNormalizedLandmark *> *> *)multiHandLandmarks

4
mediapipe/tasks/ios/test/vision/gesture_recognizer/utils/sources/MPPGestureRecognizerResult+ProtoHelpers.h
View File

@ -20,7 +20,9 @@ NS_ASSUME_NONNULL_BEGIN
+ (MPPGestureRecognizerResult *) + (MPPGestureRecognizerResult *)
gestureRecognizerResultsFromTextEncodedProtobufFileWithName:(NSString *)fileName gestureRecognizerResultsFromTextEncodedProtobufFileWithName:(NSString *)fileName
gestureLabel:(NSString *)gestureLabel; gestureLabel:(NSString *)gestureLabel
shouldRemoveZPosition:(BOOL)removeZPosition;
@end @end
NS_ASSUME_NONNULL_END NS_ASSUME_NONNULL_END

6
mediapipe/tasks/ios/test/vision/gesture_recognizer/utils/sources/MPPGestureRecognizerResult+ProtoHelpers.mm
View File

@ -33,12 +33,15 @@ using ::mediapipe::tasks::ios::test::vision::utils::get_proto_from_pbtxt;
+ (MPPGestureRecognizerResult *) + (MPPGestureRecognizerResult *)
gestureRecognizerResultsFromTextEncodedProtobufFileWithName:(NSString *)fileName gestureRecognizerResultsFromTextEncodedProtobufFileWithName:(NSString *)fileName
gestureLabel:(NSString *)gestureLabel { gestureLabel:(NSString *)gestureLabel
shouldRemoveZPosition:(BOOL)removeZPosition {
LandmarksDetectionResultProto landmarkDetectionResultProto; LandmarksDetectionResultProto landmarkDetectionResultProto;
if (!get_proto_from_pbtxt(fileName.cppString, landmarkDetectionResultProto).ok()) { if (!get_proto_from_pbtxt(fileName.cppString, landmarkDetectionResultProto).ok()) {
return nil; return nil;
} }
if (removeZPosition) {
// Remove z position of landmarks, because they are not used in correctness // Remove z position of landmarks, because they are not used in correctness
// testing. For video or live stream mode, the z positions varies a lot during // testing. For video or live stream mode, the z positions varies a lot during
// tracking from frame to frame. // tracking from frame to frame.
@ -46,6 +49,7 @@ using ::mediapipe::tasks::ios::test::vision::utils::get_proto_from_pbtxt;
auto &landmark = *landmarkDetectionResultProto.mutable_landmarks()->mutable_landmark(i); auto &landmark = *landmarkDetectionResultProto.mutable_landmarks()->mutable_landmark(i);
landmark.clear_z(); landmark.clear_z();
} }
}
ClassificationListProto gesturesProto; ClassificationListProto gesturesProto;
ClassificationProto *classificationProto = gesturesProto.add_classification(); ClassificationProto *classificationProto = gesturesProto.add_classification();