talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add TensorsToFaceLandmarksGraph to support two types of face mesh models.

Browse Source 
PiperOrigin-RevId: 511626032
This commit is contained in:
MediaPipe Team
2023-02-22 16:22:13 -08:00
committed by Copybara-Service
parent 9e7950a69a
commit 2c3c1e664a
9 changed files with 2438 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/cc/vision/face_landmarker/BUILD
+23
View File
@@ -18,6 +18,27 @@ package(default_visibility = [
licenses(["notice"])
cc_library(
name = "tensors_to_face_landmarks_graph",
srcs = ["tensors_to_face_landmarks_graph.cc"],
deps = [
"//mediapipe/calculators/core:split_vector_calculator",
"//mediapipe/calculators/core:split_vector_calculator_cc_proto",
"//mediapipe/calculators/tensor:tensors_to_landmarks_calculator",
"//mediapipe/calculators/tensor:tensors_to_landmarks_calculator_cc_proto",
"//mediapipe/calculators/util:landmarks_refinement_calculator",
"//mediapipe/calculators/util:landmarks_refinement_calculator_cc_proto",
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/framework:calculator_framework",
"//mediapipe/framework:subgraph",
"//mediapipe/framework/api2:builder",
"//mediapipe/framework/formats:landmark_cc_proto",
"//mediapipe/framework/formats:tensor",
"//mediapipe/tasks/cc/vision/face_landmarker/proto:tensors_to_face_landmarks_graph_options_cc_proto",
],
alwayslink = 1,
)
cc_library(
name = "face_blendshapes_graph",
srcs = ["face_blendshapes_graph.cc"],
@@ -48,6 +69,7 @@ cc_library(
name = "face_landmarks_detector_graph",
srcs = ["face_landmarks_detector_graph.cc"],
deps = [
":tensors_to_face_landmarks_graph",
"//mediapipe/calculators/core:begin_loop_calculator",
"//mediapipe/calculators/core:end_loop_calculator",
"//mediapipe/calculators/core:split_vector_calculator",
@@ -80,6 +102,7 @@ cc_library(
"//mediapipe/tasks/cc/core:model_task_graph",
"//mediapipe/tasks/cc/core:utils",
"//mediapipe/tasks/cc/vision/face_landmarker/proto:face_landmarks_detector_graph_options_cc_proto",
"//mediapipe/tasks/cc/vision/face_landmarker/proto:tensors_to_face_landmarks_graph_options_cc_proto",
"//mediapipe/tasks/cc/vision/utils:image_tensor_specs",
],
alwayslink = 1,