From 9bb852c33d811d0329a7e3b423f691fe4b4c8f34 Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Thu, 31 Aug 2023 17:07:08 -0700 Subject: [PATCH] Add libimagegenerator_gpu.so PiperOrigin-RevId: 561800710 --- .../tasks/vision/imagegenerator/BUILD | 7 +++++++ third_party/external_files.bzl | 18 ++++++++++++------ 2 files changed, 19 insertions(+), 6 deletions(-) diff --git a/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/imagegenerator/BUILD b/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/imagegenerator/BUILD index 5a460009a..3a55c602f 100644 --- a/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/imagegenerator/BUILD +++ b/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/imagegenerator/BUILD @@ -12,10 +12,17 @@ # See the License for the specific language governing permissions and # limitations under the License. +load( + "//mediapipe/framework/tool:mediapipe_files.bzl", + "mediapipe_files", +) + licenses(["notice"]) package(default_visibility = ["//visibility:public"]) +mediapipe_files(srcs = ["libimagegenerator_gpu.so"]) + # The native library of MediaPipe vision image generator tasks. cc_binary( name = "libmediapipe_tasks_vision_image_generator_jni.so", diff --git a/third_party/external_files.bzl b/third_party/external_files.bzl index 969a3fcac..86d504600 100644 --- a/third_party/external_files.bzl +++ b/third_party/external_files.bzl @@ -264,8 +264,8 @@ def external_files(): http_file( name = "com_google_mediapipe_dynamic_input_classifier_tflite", - sha256 = "fb34b05e1cd4081f3c2bb882092f617efb19266b3353d51b3790a172cae09784", - urls = ["https://storage.googleapis.com/mediapipe-assets/dynamic_input_classifier.tflite?generation=1680543275416843"], + sha256 = "c5499daf5773cef89ce984df329c6324194a83bea7c7cf83159bf660a58de85c", + urls = ["https://storage.googleapis.com/mediapipe-assets/dynamic_input_classifier.tflite?generation=1693433004555536"], ) http_file( @@ -282,8 +282,8 @@ def external_files(): http_file( name = "com_google_mediapipe_efficientdet_lite0_fp16_no_nms_tflite", - sha256 = "237a58389081333e5cf4154e42b593ce7dd357445536fcaf4ca5bc51c2c50f1c", - urls = ["https://storage.googleapis.com/mediapipe-assets/efficientdet_lite0_fp16_no_nms.tflite?generation=1682632067597216"], + sha256 = "bcda125c96d3767bca894c8cbe7bc458379c9974c9fd8bdc6204e7124a74082a", + urls = ["https://storage.googleapis.com/mediapipe-assets/efficientdet_lite0_fp16_no_nms.tflite?generation=1693433007348701"], ) http_file( @@ -432,8 +432,8 @@ def external_files(): http_file( name = "com_google_mediapipe_face_stylizer_task", - sha256 = "b34f3896cbe860468538cf5a562c0468964f182b8bb07cb527224312969d1625", - urls = ["https://storage.googleapis.com/mediapipe-assets/face_stylizer.task?generation=1682627841126340"], + sha256 = "423f350aab236123818adb7b39e0a14e14708a9a019fb2fe00a015a2561fd0c8", + urls = ["https://storage.googleapis.com/mediapipe-assets/face_stylizer.task?generation=1693433010526766"], ) http_file( @@ -664,6 +664,12 @@ def external_files(): urls = ["https://storage.googleapis.com/mediapipe-assets/leopard.jpg?generation=1685997280368627"], ) + http_file( + name = "com_google_mediapipe_libimagegenerator_gpu_so", + sha256 = "39ed9738297fa051a7f3cc9bdb7189418a9e118aa3cad4e1d577995837fdd58c", + urls = ["https://storage.googleapis.com/mediapipe-assets/libimagegenerator_gpu.so?generation=1693433013917189"], + ) + http_file( name = "com_google_mediapipe_mobilebert_embedding_with_metadata_tflite", sha256 = "fa47142dcc6f446168bc672f2df9605b6da5d0c0d6264e9be62870282365b95c",