Added MPPGestureRecognizerOptionsHelpers

This commit is contained in:
Prianka Liz Kariat 2023-05-17 21:51:40 +05:30
parent a4c280310b
commit ebd1545506
3 changed files with 148 additions and 0 deletions

View File

@ -0,0 +1,33 @@
# Copyright 2023 The MediaPipe Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
package(default_visibility = ["//mediapipe/tasks:internal"])
licenses(["notice"])
objc_library(
name = "MPPGestureRecognizerOptionsHelpers",
srcs = ["sources/MPPGestureRecognizerOptions+Helpers.mm"],
hdrs = ["sources/MPPGestureRecognizerOptions+Helpers.h"],
deps = [
"//mediapipe/framework:calculator_options_cc_proto",
"//mediapipe/tasks/cc/vision/gesture_recognizer/proto:gesture_recognizer_graph_options_cc_proto",
"//mediapipe/tasks/ios/common/utils:NSStringHelpers",
"//mediapipe/tasks/ios/components/processors/utils:MPPClassifierOptionsHelpers",
"//mediapipe/tasks/ios/core:MPPTaskOptionsProtocol",
"//mediapipe/tasks/ios/core/utils:MPPBaseOptionsHelpers",
"//mediapipe/tasks/ios/vision/gesture_recognizer:MPPGestureRecognizerOptions",
],
)

View File

@ -0,0 +1,32 @@
// Copyright 2023 The MediaPipe Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "mediapipe/framework/calculator_options.pb.h"
#import "mediapipe/tasks/ios/core/sources/MPPTaskOptionsProtocol.h"
#import "mediapipe/tasks/ios/vision/gesture_recognizer/sources/MPPGestureRecognizerOptions.h"
NS_ASSUME_NONNULL_BEGIN
@interface MPPGestureRecognizerOptions (Helpers) <MPPTaskOptionsProtocol>
/**
* Populates the provided `CalculatorOptions` proto container with the current settings.
*
* @param optionsProto The `CalculatorOptions` proto object to copy the settings to.
*/
- (void)copyToProto:(::mediapipe::CalculatorOptions *)optionsProto;
@end
NS_ASSUME_NONNULL_END

View File

@ -0,0 +1,83 @@
// Copyright 2023 The MediaPipe Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#import "mediapipe/tasks/ios/vision/gesture_recognizer/utils/sources/MPPGestureRecognizerOptions+Helpers.h"
#import "mediapipe/tasks/ios/common/utils/sources/NSString+Helpers.h"
#import "mediapipe/tasks/ios/components/processors/utils/sources/MPPClassifierOptions+Helpers.h"
#import "mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.h"
#include "mediapipe/tasks/cc/vision/gesture_recognizer/proto/gesture_recognizer_graph_options.pb.h"
namespace {
using CalculatorOptionsProto = mediapipe::CalculatorOptions;
using GestureClassifierGraphOptionsProto =
::mediapipe::tasks::vision::gesture_recognizer::proto::GestureClassifierGraphOptions;
using GestureRecognizerGraphOptionsProto =
::mediapipe::tasks::vision::gesture_recognizer::proto::GestureRecognizerGraphOptions;
using HandGestureRecognizerGraphOptionsProto =
::mediapipe::tasks::vision::gesture_recognizer::proto::HandGestureRecognizerGraphOptions;
using HandLandmarkerGraphOptionsProto =
::mediapipe::tasks::vision::hand_landmarker::proto::HandLandmarkerGraphOptions;
using HandDetectorGraphOptionsProto =
::mediapipe::tasks::vision::hand_detector::proto::HandDetectorGraphOptions;
using HandLandmarksDetectorGraphOptionsProto =
::mediapipe::tasks::vision::hand_landmarker::proto::HandLandmarksDetectorGraphOptions;
using ClassifierOptionsProto = ::mediapipe::tasks::components::processors::proto::ClassifierOptions;
} // namespace
@implementation MPPGestureRecognizerOptions (Helpers)
- (void)copyToProto:(CalculatorOptionsProto *)optionsProto {
GestureRecognizerGraphOptionsProto *gestureRecognizerGraphOptionsProto =
optionsProto->MutableExtension(GestureRecognizerGraphOptionsProto::ext);
[self.baseOptions copyToProto:gestureRecognizerGraphOptionsProto->mutable_base_options()
withUseStreamMode:self.runningMode != MPPRunningModeImage];
HandLandmarkerGraphOptionsProto *handLandmarkerGraphOptionsProto =
gestureRecognizerGraphOptionsProto->mutable_hand_landmarker_graph_options();
handLandmarkerGraphOptionsProto->set_min_tracking_confidence(self.minTrackingConfidence);
HandDetectorGraphOptionsProto *handDetectorGraphOptionsProto =
handLandmarkerGraphOptionsProto->mutable_hand_detector_graph_options();
handDetectorGraphOptionsProto->Clear();
handDetectorGraphOptionsProto->set_num_hands(self.numberOfHands);
handDetectorGraphOptionsProto->set_min_detection_confidence(self.minHandDetectionConfidence);
HandLandmarksDetectorGraphOptionsProto *handLandmarksDetectorGraphOptionsProto =
handLandmarkerGraphOptionsProto->mutable_hand_landmarks_detector_graph_options();
handLandmarksDetectorGraphOptionsProto->Clear();
handLandmarksDetectorGraphOptionsProto->set_min_detection_confidence(
self.minHandPresenceConfidence);
HandGestureRecognizerGraphOptionsProto *handGestureRecognizerGraphOptionsProto =
gestureRecognizerGraphOptionsProto->mutable_hand_gesture_recognizer_graph_options();
if (self.cannedGesturesClassifierOptions) {
GestureClassifierGraphOptionsProto *cannedGesturesClassifierOptionsProto =
handGestureRecognizerGraphOptionsProto->mutable_canned_gesture_classifier_graph_options();
[self.cannedGesturesClassifierOptions
copyToProto:cannedGesturesClassifierOptionsProto->mutable_classifier_options()];
}
if (self.customGesturesClassifierOptions) {
GestureClassifierGraphOptionsProto *customGesturesClassifierOptionsProto =
handGestureRecognizerGraphOptionsProto->mutable_custom_gesture_classifier_graph_options();
[self.customGesturesClassifierOptions
copyToProto:customGesturesClassifierOptionsProto->mutable_classifier_options()];
}
}
@end