build: include blaze_face_short_range.tflite

homuler · Jul 29, 2023 · 20d4527 · 20d4527
1 parent 216a453
commit 20d4527
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/...com.github.homuler.mediapipe/PackageResources/MediaPipe/blaze_face_short_range.bytes.meta b/...com.github.homuler.mediapipe/PackageResources/MediaPipe/blaze_face_short_range.bytes.meta
diff --git a/mediapipe_api/BUILD b/mediapipe_api/BUILD
@@ -9,6 +9,7 @@ load("@bazel_skylib//lib:selects.bzl", "selects")
 load("@bazel_skylib//rules:common_settings.bzl", "bool_flag", "string_list_flag")
 load("@rules_pkg//:pkg.bzl", "pkg_zip")
 load("@build_bazel_apple_support//rules:universal_binary.bzl", "universal_binary")
+load("@com_google_mediapipe//mediapipe/framework/tool:mediapipe_files.bzl", "mediapipe_files")
 
 bool_flag(
     name = "macos_universal",
@@ -427,6 +428,10 @@ pkg_zip(
     ],
 )
 
+mediapipe_files(srcs = [
+    "blaze_face_short_range.tflite",
+])
+
 pkg_asset(
     name = "mediapipe_assets",
     srcs = select({
@@ -466,6 +471,7 @@ pkg_asset(
 filegroup(
     name = "face_detection_assets",
     srcs = [
+        ":blaze_face_short_range.tflite",
         "@com_google_mediapipe//mediapipe/modules/face_detection:face_detection_full_range_sparse.tflite",
         "@com_google_mediapipe//mediapipe/modules/face_detection:face_detection_full_range.tflite",
         "@com_google_mediapipe//mediapipe/modules/face_detection:face_detection_short_range.tflite",

diff --git a/third_party/mediapipe_model_path.diff b/third_party/mediapipe_model_path.diff
@@ -487,3 +487,20 @@ index 00000000..946be0d4
 +
 +}  // namespace internal
 +}  // namespace mediapipe
+diff --git a/third_party/external_files.bzl b/third_party/external_files.bzl
+index 652a2947..c8e1a1ea 100644
+--- a/third_party/external_files.bzl
++++ b/third_party/external_files.bzl
+@@ -370,6 +370,12 @@ def external_files():
+         urls = ["https://storage.googleapis.com/mediapipe-assets/face_detection_short_range.tflite?generation=1677044301978921"],
+     )
+
++    http_file(
++        name = "com_google_mediapipe_blaze_face_short_range_tflite",
++        sha256 = "b4578f35940bf5a1a655214a1cce5cab13eba73c1297cd78e1a04c2380b0152f",
++        urls = ["https://storage.googleapis.com/mediapipe-models/face_detector/blaze_face_short_range/float16/1/blaze_face_short_range.tflite?generation=1682480001338381"],
++    )
++
+     http_file(
+         name = "com_google_mediapipe_face_geometry_expected_out_pbtxt",
+         sha256 = "4a4ed08055a5bc9281472dd60180d11f0cdc9a15fa1788d87a58af3d06b2c6e4",