From 823493ee8226c0dc52f36f196fa4ac6145d27b3c Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Fri, 1 Sep 2023 10:50:36 -0700 Subject: [PATCH] Internal update. PiperOrigin-RevId: 561995055 --- .../com/google/mediapipe/tasks/vision/BUILD | 21 +++++++++++++++++++ .../tasks/vision/imagegenerator/BUILD | 13 +++++++++--- 2 files changed, 31 insertions(+), 3 deletions(-) diff --git a/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/BUILD b/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/BUILD index 1ddcd46c..60a9806e 100644 --- a/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/BUILD +++ b/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/BUILD @@ -44,6 +44,13 @@ android_library( ], ) +# Expose the java source files for building mediapipe tasks vision per task AAR. +filegroup( + name = "core_java_src", + srcs = glob(["core/**/*.java"]), + visibility = ["//mediapipe/tasks/java/com/google/mediapipe/tasks/vision:__subpackages__"], +) + # The native library of all MediaPipe vision tasks. cc_binary( name = "libmediapipe_tasks_vision_jni.so", @@ -294,6 +301,13 @@ android_library( ], ) +# Expose the java source files for building mediapipe tasks vision per task AAR. +filegroup( + name = "image_segmenter_java_src", + srcs = glob(["imagesegmenter/**/*.java"]), + visibility = ["//mediapipe/tasks/java/com/google/mediapipe/tasks/vision:__subpackages__"], +) + android_library( name = "interactivesegmenter", srcs = [ @@ -409,6 +423,13 @@ android_library( ], ) +# Expose the java source files for building mediapipe tasks vision per task AAR. +filegroup( + name = "face_landmarker_java_src", + srcs = glob(["facelandmarker/**/*.java"]), + visibility = ["//mediapipe/tasks/java/com/google/mediapipe/tasks/vision:__subpackages__"], +) + load("//mediapipe/tasks/java/com/google/mediapipe/tasks:mediapipe_tasks_aar.bzl", "mediapipe_tasks_vision_aar") mediapipe_tasks_vision_aar( 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 3a55c602..5c4bb3f9 100644 --- a/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/imagegenerator/BUILD +++ b/mediapipe/tasks/java/com/google/mediapipe/tasks/vision/imagegenerator/BUILD @@ -44,8 +44,11 @@ cc_binary( ) cc_library( - name = "libmediapipe_tasks_vision_image_generator_jni_lib", - srcs = [":libmediapipe_tasks_vision_image_generator_jni.so"], + name = "mediapipe_tasks_vision_image_generator_jni_lib", + srcs = [ + ":libimagegenerator_gpu.so", + ":libmediapipe_tasks_vision_image_generator_jni.so", + ], alwayslink = 1, ) @@ -86,6 +89,10 @@ load("//mediapipe/tasks/java/com/google/mediapipe/tasks:mediapipe_tasks_aar.bzl" mediapipe_tasks_vision_image_generator_aar( name = "tasks_vision_image_generator", - srcs = glob(["**/*.java"]), + srcs = glob(["**/*.java"]) + [ + "//mediapipe/tasks/java/com/google/mediapipe/tasks/vision:core_java_src", + "//mediapipe/tasks/java/com/google/mediapipe/tasks/vision:face_landmarker_java_src", + "//mediapipe/tasks/java/com/google/mediapipe/tasks/vision:image_segmenter_java_src", + ], native_library = ":libmediapipe_tasks_vision_image_generator_jni_lib", )