Create MediaPipe "tasks-text" AAR.
PiperOrigin-RevId: 484004494
This commit is contained in:
parent
254f786624
commit
a941c5cdd1
|
@ -40,6 +40,10 @@ _VISION_TASKS_JAVA_PROTO_LITE_TARGETS = [
|
||||||
"//mediapipe/tasks/cc/vision/hand_landmarker/proto:hand_landmarks_detector_graph_options_java_proto_lite",
|
"//mediapipe/tasks/cc/vision/hand_landmarker/proto:hand_landmarks_detector_graph_options_java_proto_lite",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
_TEXT_TASKS_JAVA_PROTO_LITE_TARGETS = [
|
||||||
|
"//mediapipe/tasks/cc/text/text_classifier/proto:text_classifier_graph_options_java_proto_lite",
|
||||||
|
]
|
||||||
|
|
||||||
def mediapipe_tasks_core_aar(name, srcs, manifest):
|
def mediapipe_tasks_core_aar(name, srcs, manifest):
|
||||||
"""Builds medaipipe tasks core AAR.
|
"""Builds medaipipe tasks core AAR.
|
||||||
|
|
||||||
|
@ -60,6 +64,11 @@ def mediapipe_tasks_core_aar(name, srcs, manifest):
|
||||||
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
for target in _TEXT_TASKS_JAVA_PROTO_LITE_TARGETS:
|
||||||
|
mediapipe_tasks_java_proto_srcs.append(
|
||||||
|
_mediapipe_tasks_java_proto_src_extractor(target = target),
|
||||||
|
)
|
||||||
|
|
||||||
mediapipe_tasks_java_proto_srcs.append(mediapipe_java_proto_src_extractor(
|
mediapipe_tasks_java_proto_srcs.append(mediapipe_java_proto_src_extractor(
|
||||||
target = "//mediapipe/calculators/core:flow_limiter_calculator_java_proto_lite",
|
target = "//mediapipe/calculators/core:flow_limiter_calculator_java_proto_lite",
|
||||||
src_out = "com/google/mediapipe/calculator/proto/FlowLimiterCalculatorProto.java",
|
src_out = "com/google/mediapipe/calculator/proto/FlowLimiterCalculatorProto.java",
|
||||||
|
@ -106,7 +115,10 @@ def mediapipe_tasks_core_aar(name, srcs, manifest):
|
||||||
"@maven//:com_google_flogger_flogger",
|
"@maven//:com_google_flogger_flogger",
|
||||||
"@maven//:com_google_flogger_flogger_system_backend",
|
"@maven//:com_google_flogger_flogger_system_backend",
|
||||||
"@maven//:com_google_code_findbugs_jsr305",
|
"@maven//:com_google_code_findbugs_jsr305",
|
||||||
] + _CORE_TASKS_JAVA_PROTO_LITE_TARGETS + _VISION_TASKS_JAVA_PROTO_LITE_TARGETS,
|
] +
|
||||||
|
_CORE_TASKS_JAVA_PROTO_LITE_TARGETS +
|
||||||
|
_VISION_TASKS_JAVA_PROTO_LITE_TARGETS +
|
||||||
|
_TEXT_TASKS_JAVA_PROTO_LITE_TARGETS,
|
||||||
)
|
)
|
||||||
|
|
||||||
def mediapipe_tasks_vision_aar(name, srcs, native_library):
|
def mediapipe_tasks_vision_aar(name, srcs, native_library):
|
||||||
|
@ -142,6 +154,39 @@ EOF
|
||||||
native_library = native_library,
|
native_library = native_library,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def mediapipe_tasks_text_aar(name, srcs, native_library):
|
||||||
|
"""Builds medaipipe tasks text AAR.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
name: The bazel target name.
|
||||||
|
srcs: MediaPipe Text Tasks' source files.
|
||||||
|
native_library: The native library that contains text tasks' graph and calculators.
|
||||||
|
"""
|
||||||
|
|
||||||
|
native.genrule(
|
||||||
|
name = name + "tasks_manifest_generator",
|
||||||
|
outs = ["AndroidManifest.xml"],
|
||||||
|
cmd = """
|
||||||
|
cat > $(OUTS) <<EOF
|
||||||
|
<?xml version="1.0" encoding="utf-8"?>
|
||||||
|
<manifest xmlns:android="http://schemas.android.com/apk/res/android"
|
||||||
|
package="com.google.mediapipe.tasks.text">
|
||||||
|
<uses-sdk
|
||||||
|
android:minSdkVersion="24"
|
||||||
|
android:targetSdkVersion="30" />
|
||||||
|
</manifest>
|
||||||
|
EOF
|
||||||
|
""",
|
||||||
|
)
|
||||||
|
|
||||||
|
_mediapipe_tasks_aar(
|
||||||
|
name = name,
|
||||||
|
srcs = srcs,
|
||||||
|
manifest = "AndroidManifest.xml",
|
||||||
|
java_proto_lite_targets = _CORE_TASKS_JAVA_PROTO_LITE_TARGETS + _TEXT_TASKS_JAVA_PROTO_LITE_TARGETS,
|
||||||
|
native_library = native_library,
|
||||||
|
)
|
||||||
|
|
||||||
def _mediapipe_tasks_aar(name, srcs, manifest, java_proto_lite_targets, native_library):
|
def _mediapipe_tasks_aar(name, srcs, manifest, java_proto_lite_targets, native_library):
|
||||||
"""Builds medaipipe tasks AAR."""
|
"""Builds medaipipe tasks AAR."""
|
||||||
|
|
||||||
|
|
|
@ -61,3 +61,11 @@ android_library(
|
||||||
"@maven//:com_google_guava_guava",
|
"@maven//:com_google_guava_guava",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
load("//mediapipe/tasks/java/com/google/mediapipe/tasks:mediapipe_tasks_aar.bzl", "mediapipe_tasks_text_aar")
|
||||||
|
|
||||||
|
mediapipe_tasks_text_aar(
|
||||||
|
name = "tasks_text",
|
||||||
|
srcs = glob(["**/*.java"]),
|
||||||
|
native_library = ":libmediapipe_tasks_text_jni_lib",
|
||||||
|
)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user