From 7a9be52c7b0548a5f9766b6b948ded7e92752b57 Mon Sep 17 00:00:00 2001 From: "chopper.lee" Date: Tue, 4 Apr 2023 13:57:39 +0900 Subject: [PATCH] fix: specify missing tflite model After being updated to download tflite files from gcp, the necessary tflite files have to be specified in BUILD. In the face mesh example, the tflite file required for face detection was missing, so it was specified. --- mediapipe/examples/desktop/face_mesh/BUILD | 15 ++++++++++++--- .../examples/desktop/holistic_tracking/BUILD | 2 ++ 2 files changed, 14 insertions(+), 3 deletions(-) diff --git a/mediapipe/examples/desktop/face_mesh/BUILD b/mediapipe/examples/desktop/face_mesh/BUILD index 162cb6b65..e87d5302b 100644 --- a/mediapipe/examples/desktop/face_mesh/BUILD +++ b/mediapipe/examples/desktop/face_mesh/BUILD @@ -18,7 +18,10 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"]) cc_binary( name = "face_mesh_tflite", - data = ["//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite"], + data = [ + "//mediapipe/modules/face_detection:face_detection_short_range.tflite", + "//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite", + ], deps = [ "//mediapipe/examples/desktop:simple_run_graph_main", "//mediapipe/graphs/face_mesh:desktop_calculators", @@ -27,7 +30,10 @@ cc_binary( cc_binary( name = "face_mesh_cpu", - data = ["//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite"], + data = [ + "//mediapipe/modules/face_detection:face_detection_short_range.tflite", + "//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite", + ], deps = [ "//mediapipe/examples/desktop:demo_run_graph_main", "//mediapipe/graphs/face_mesh:desktop_live_calculators", @@ -37,7 +43,10 @@ cc_binary( # Linux only cc_binary( name = "face_mesh_gpu", - data = ["//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite"], + data = [ + "//mediapipe/modules/face_detection:face_detection_short_range.tflite", + "//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite", + ], deps = [ "//mediapipe/examples/desktop:demo_run_graph_main_gpu", "//mediapipe/graphs/face_mesh:desktop_live_gpu_calculators", diff --git a/mediapipe/examples/desktop/holistic_tracking/BUILD b/mediapipe/examples/desktop/holistic_tracking/BUILD index bb5d747cc..a3c802959 100644 --- a/mediapipe/examples/desktop/holistic_tracking/BUILD +++ b/mediapipe/examples/desktop/holistic_tracking/BUILD @@ -19,6 +19,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"]) cc_binary( name = "holistic_tracking_cpu", data = [ + "//mediapipe/modules/face_detection:face_detection_short_range.tflite", "//mediapipe/modules/face_landmark:face_landmark.tflite", "//mediapipe/modules/hand_landmark:hand_landmark_full.tflite", "//mediapipe/modules/holistic_landmark:hand_recrop.tflite", @@ -35,6 +36,7 @@ cc_binary( cc_binary( name = "holistic_tracking_gpu", data = [ + "//mediapipe/modules/face_detection:face_detection_short_range.tflite", "//mediapipe/modules/face_landmark:face_landmark.tflite", "//mediapipe/modules/hand_landmark:hand_landmark_full.tflite", "//mediapipe/modules/holistic_landmark:hand_recrop.tflite",