Update Gesture Recognizer Graph Options protos.
PiperOrigin-RevId: 479141569
This commit is contained in:
parent
668f711b71
commit
d3b66ba253
|
@ -102,8 +102,7 @@ Source<std::vector<Tensor>> ConvertMatrixToTensor(Source<Matrix> matrix,
|
||||||
//
|
//
|
||||||
// Example:
|
// Example:
|
||||||
// node {
|
// node {
|
||||||
// calculator:
|
// calculator: "mediapipe.tasks.vision.SingleHandGestureRecognizerGraph"
|
||||||
// "mediapipe.tasks.vision.gesture_recognizer.SingleHandGestureRecognizerGraph"
|
|
||||||
// input_stream: "HANDEDNESS:handedness"
|
// input_stream: "HANDEDNESS:handedness"
|
||||||
// input_stream: "LANDMARKS:landmarks"
|
// input_stream: "LANDMARKS:landmarks"
|
||||||
// input_stream: "WORLD_LANDMARKS:world_landmarks"
|
// input_stream: "WORLD_LANDMARKS:world_landmarks"
|
||||||
|
|
|
@ -45,9 +45,23 @@ mediapipe_proto_library(
|
||||||
name = "hand_gesture_recognizer_graph_options_proto",
|
name = "hand_gesture_recognizer_graph_options_proto",
|
||||||
srcs = ["hand_gesture_recognizer_graph_options.proto"],
|
srcs = ["hand_gesture_recognizer_graph_options.proto"],
|
||||||
deps = [
|
deps = [
|
||||||
|
":gesture_classifier_graph_options_proto",
|
||||||
|
":gesture_embedder_graph_options_proto",
|
||||||
"//mediapipe/framework:calculator_options_proto",
|
"//mediapipe/framework:calculator_options_proto",
|
||||||
"//mediapipe/framework:calculator_proto",
|
"//mediapipe/framework:calculator_proto",
|
||||||
"//mediapipe/tasks/cc/components/processors/proto:classifier_options_proto",
|
"//mediapipe/tasks/cc/components/processors/proto:classifier_options_proto",
|
||||||
"//mediapipe/tasks/cc/core/proto:base_options_proto",
|
"//mediapipe/tasks/cc/core/proto:base_options_proto",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
mediapipe_proto_library(
|
||||||
|
name = "gesture_recognizer_graph_options_proto",
|
||||||
|
srcs = ["gesture_recognizer_graph_options.proto"],
|
||||||
|
deps = [
|
||||||
|
":hand_gesture_recognizer_graph_options_proto",
|
||||||
|
"//mediapipe/framework:calculator_options_proto",
|
||||||
|
"//mediapipe/framework:calculator_proto",
|
||||||
|
"//mediapipe/tasks/cc/core/proto:base_options_proto",
|
||||||
|
"//mediapipe/tasks/cc/vision/hand_landmarker/proto:hand_landmarker_graph_options_proto",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
|
@ -0,0 +1,40 @@
|
||||||
|
/* Copyright 2022 The MediaPipe Authors. All Rights Reserved.
|
||||||
|
|
||||||
|
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.
|
||||||
|
==============================================================================*/
|
||||||
|
|
||||||
|
syntax = "proto2";
|
||||||
|
|
||||||
|
package mediapipe.tasks.vision.gesture_recognizer.proto;
|
||||||
|
|
||||||
|
import "mediapipe/framework/calculator.proto";
|
||||||
|
import "mediapipe/tasks/cc/core/proto/base_options.proto";
|
||||||
|
import "mediapipe/tasks/cc/vision/gesture_recognizer/proto/hand_gesture_recognizer_graph_options.proto";
|
||||||
|
import "mediapipe/tasks/cc/vision/hand_landmarker/proto/hand_landmarker_graph_options.proto";
|
||||||
|
|
||||||
|
message GestureRecognizerGraphOptions {
|
||||||
|
extend mediapipe.CalculatorOptions {
|
||||||
|
optional GestureRecognizerGraphOptions ext = 478371831;
|
||||||
|
}
|
||||||
|
// Base options for configuring gesture recognizer graph, such as specifying
|
||||||
|
// the TfLite model file with metadata, accelerator options, etc.
|
||||||
|
optional core.proto.BaseOptions base_options = 1;
|
||||||
|
|
||||||
|
// Options for configuring hand landmarker subgraph.
|
||||||
|
optional hand_landmarker.proto.HandLandmarkerGraphOptions
|
||||||
|
hand_landmarker_graph_options = 2;
|
||||||
|
|
||||||
|
// Options for configuring hand gesture recognizer graph.
|
||||||
|
optional HandGestureRecognizerGraphOptions
|
||||||
|
hand_gesture_recognizer_graph_options = 3;
|
||||||
|
}
|
|
@ -20,6 +20,8 @@ package mediapipe.tasks.vision.gesture_recognizer.proto;
|
||||||
import "mediapipe/framework/calculator.proto";
|
import "mediapipe/framework/calculator.proto";
|
||||||
import "mediapipe/tasks/cc/components/processors/proto/classifier_options.proto";
|
import "mediapipe/tasks/cc/components/processors/proto/classifier_options.proto";
|
||||||
import "mediapipe/tasks/cc/core/proto/base_options.proto";
|
import "mediapipe/tasks/cc/core/proto/base_options.proto";
|
||||||
|
import "mediapipe/tasks/cc/vision/gesture_recognizer/proto/gesture_classifier_graph_options.proto";
|
||||||
|
import "mediapipe/tasks/cc/vision/gesture_recognizer/proto/gesture_embedder_graph_options.proto";
|
||||||
|
|
||||||
message HandGestureRecognizerGraphOptions {
|
message HandGestureRecognizerGraphOptions {
|
||||||
extend mediapipe.CalculatorOptions {
|
extend mediapipe.CalculatorOptions {
|
||||||
|
@ -29,11 +31,18 @@ message HandGestureRecognizerGraphOptions {
|
||||||
// specifying the TfLite model file with metadata, accelerator options, etc.
|
// specifying the TfLite model file with metadata, accelerator options, etc.
|
||||||
optional core.proto.BaseOptions base_options = 1;
|
optional core.proto.BaseOptions base_options = 1;
|
||||||
|
|
||||||
// Options for configuring the gesture classifier behavior, such as score
|
// Options for GestureEmbedder.
|
||||||
// threshold, number of results, etc.
|
optional GestureEmbedderGraphOptions gesture_embedder_graph_options = 2;
|
||||||
optional components.processors.proto.ClassifierOptions classifier_options = 2;
|
|
||||||
|
|
||||||
// Minimum confidence value ([0.0, 1.0]) for the hand landmarks to be
|
// Options for GestureClassifier of default gestures.
|
||||||
// considered tracked successfully
|
optional GestureClassifierGraphOptions
|
||||||
optional float min_tracking_confidence = 3 [default = 0.0];
|
canned_gesture_classifier_graph_options = 3;
|
||||||
|
|
||||||
|
// Options for GestureClassifier of custom gestures.
|
||||||
|
optional GestureClassifierGraphOptions
|
||||||
|
custom_gesture_classifier_graph_options = 4;
|
||||||
|
|
||||||
|
// TODO: remove these. Temporary solutions before bundle asset is
|
||||||
|
// ready.
|
||||||
|
optional components.processors.proto.ClassifierOptions classifier_options = 5;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user