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

Add necessary java lite proto source code to mediapipe tasks aar for the face landmarker task.

Browse Source 
PiperOrigin-RevId: 520143851
This commit is contained in:
Jiuqiang Tang 2023-03-28 14:59:26 -07:00 committed by Copybara-Service
parent 62d86494b0
commit 1e77725a15
4 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mediapipe/tasks/cc/vision/face_geometry/calculators/geometry_pipeline_calculator.proto
View File

@ -19,6 +19,9 @@ package mediapipe.tasks.vision.face_geometry;
import "mediapipe/framework/calculator_options.proto"; import "mediapipe/framework/calculator_options.proto";
import "mediapipe/tasks/cc/core/proto/external_file.proto"; import "mediapipe/tasks/cc/core/proto/external_file.proto";
option java_package = "com.google.mediapipe.tasks.vision.facegeometry.calculators.proto";
option java_outer_classname = "FaceGeometryPipelineCalculatorOptionsProto";
message FaceGeometryPipelineCalculatorOptions { message FaceGeometryPipelineCalculatorOptions {
extend mediapipe.CalculatorOptions { extend mediapipe.CalculatorOptions {
optional FaceGeometryPipelineCalculatorOptions ext = 512499200; optional FaceGeometryPipelineCalculatorOptions ext = 512499200;

3
mediapipe/tasks/cc/vision/face_geometry/proto/face_geometry_graph_options.proto
View File

@ -19,6 +19,9 @@ package mediapipe.tasks.vision.face_geometry.proto;
import "mediapipe/framework/calculator_options.proto"; import "mediapipe/framework/calculator_options.proto";
import "mediapipe/tasks/cc/vision/face_geometry/calculators/geometry_pipeline_calculator.proto"; import "mediapipe/tasks/cc/vision/face_geometry/calculators/geometry_pipeline_calculator.proto";
option java_package = "com.google.mediapipe.tasks.vision.facegeometry.proto";
option java_outer_classname = "FaceGeometryGraphOptionsProto";
message FaceGeometryGraphOptions { message FaceGeometryGraphOptions {
extend mediapipe.CalculatorOptions { extend mediapipe.CalculatorOptions {
optional FaceGeometryGraphOptions ext = 515723506; optional FaceGeometryGraphOptions ext = 515723506;

2
mediapipe/tasks/cc/vision/face_geometry/proto/mesh_3d.proto
View File

@ -17,7 +17,7 @@ syntax = "proto2";
package mediapipe.tasks.vision.face_geometry.proto; package mediapipe.tasks.vision.face_geometry.proto;
option java_package = "com.google.mediapipe.tasks.vision.facegeometry.proto"; option java_package = "com.google.mediapipe.tasks.vision.facegeometry.proto";
option java_outer_classname = "Mesh3dProto"; option java_outer_classname = "Mesh3DProto";
message Mesh3d { message Mesh3d {
enum VertexType { enum VertexType {

8
mediapipe/tasks/java/com/google/mediapipe/tasks/mediapipe_tasks_aar.bzl
View File

@ -35,7 +35,10 @@ _AUDIO_TASKS_JAVA_PROTO_LITE_TARGETS = [
_VISION_TASKS_JAVA_PROTO_LITE_TARGETS = [ _VISION_TASKS_JAVA_PROTO_LITE_TARGETS = [
"//mediapipe/tasks/cc/vision/face_detector/proto:face_detector_graph_options_java_proto_lite", "//mediapipe/tasks/cc/vision/face_detector/proto:face_detector_graph_options_java_proto_lite",
"//mediapipe/tasks/cc/vision/face_geometry/proto:face_geometry_graph_options_java_proto_lite",
"//mediapipe/tasks/cc/vision/face_geometry/proto:face_geometry_java_proto_lite", "//mediapipe/tasks/cc/vision/face_geometry/proto:face_geometry_java_proto_lite",
"//mediapipe/tasks/cc/vision/face_geometry/proto:mesh_3d_java_proto_lite",
"//mediapipe/tasks/cc/vision/face_landmarker/proto:face_blendshapes_graph_options_java_proto_lite",
"//mediapipe/tasks/cc/vision/face_landmarker/proto:face_landmarker_graph_options_java_proto_lite", "//mediapipe/tasks/cc/vision/face_landmarker/proto:face_landmarker_graph_options_java_proto_lite",
"//mediapipe/tasks/cc/vision/face_landmarker/proto:face_landmarks_detector_graph_options_java_proto_lite", "//mediapipe/tasks/cc/vision/face_landmarker/proto:face_landmarks_detector_graph_options_java_proto_lite",
"//mediapipe/tasks/cc/vision/gesture_recognizer/proto:gesture_classifier_graph_options_java_proto_lite", "//mediapipe/tasks/cc/vision/gesture_recognizer/proto:gesture_classifier_graph_options_java_proto_lite",
@ -113,6 +116,11 @@ def mediapipe_tasks_core_aar(name, srcs, manifest):
src_out = "com/google/mediapipe/tasks/TensorsToSegmentationCalculatorOptionsProto.java", src_out = "com/google/mediapipe/tasks/TensorsToSegmentationCalculatorOptionsProto.java",
)) ))
mediapipe_tasks_java_proto_srcs.append(mediapipe_java_proto_src_extractor(
target = "//mediapipe/tasks/cc/vision/face_geometry/calculators:geometry_pipeline_calculator_java_proto_lite",
src_out = "com/google/mediapipe/tasks/vision/facegeometry/calculators/proto/FaceGeometryPipelineCalculatorOptionsProto.java",
))
android_library( android_library(
name = name, name = name,
srcs = srcs + [ srcs = srcs + [