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.
This commit is contained in:
chopper.lee 2023-04-04 13:57:39 +09:00
parent c31a4681e5
commit 7a9be52c7b
2 changed files with 14 additions and 3 deletions

View File

@ -18,7 +18,10 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary( cc_binary(
name = "face_mesh_tflite", 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 = [ deps = [
"//mediapipe/examples/desktop:simple_run_graph_main", "//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/face_mesh:desktop_calculators", "//mediapipe/graphs/face_mesh:desktop_calculators",
@ -27,7 +30,10 @@ cc_binary(
cc_binary( cc_binary(
name = "face_mesh_cpu", 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 = [ deps = [
"//mediapipe/examples/desktop:demo_run_graph_main", "//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/face_mesh:desktop_live_calculators", "//mediapipe/graphs/face_mesh:desktop_live_calculators",
@ -37,7 +43,10 @@ cc_binary(
# Linux only # Linux only
cc_binary( cc_binary(
name = "face_mesh_gpu", 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 = [ deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu", "//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/face_mesh:desktop_live_gpu_calculators", "//mediapipe/graphs/face_mesh:desktop_live_gpu_calculators",

View File

@ -19,6 +19,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary( cc_binary(
name = "holistic_tracking_cpu", name = "holistic_tracking_cpu",
data = [ data = [
"//mediapipe/modules/face_detection:face_detection_short_range.tflite",
"//mediapipe/modules/face_landmark:face_landmark.tflite", "//mediapipe/modules/face_landmark:face_landmark.tflite",
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite", "//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/holistic_landmark:hand_recrop.tflite", "//mediapipe/modules/holistic_landmark:hand_recrop.tflite",
@ -35,6 +36,7 @@ cc_binary(
cc_binary( cc_binary(
name = "holistic_tracking_gpu", name = "holistic_tracking_gpu",
data = [ data = [
"//mediapipe/modules/face_detection:face_detection_short_range.tflite",
"//mediapipe/modules/face_landmark:face_landmark.tflite", "//mediapipe/modules/face_landmark:face_landmark.tflite",
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite", "//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/holistic_landmark:hand_recrop.tflite", "//mediapipe/modules/holistic_landmark:hand_recrop.tflite",