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

fix: specify missing tflite model

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mediapipe/examples/desktop/face_mesh/BUILD
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",

2
mediapipe/examples/desktop/holistic_tracking/BUILD
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",