Implement the Java API for image classifier

PiperOrigin-RevId: 320294241

tensorflow_lite_support/cc/task/vision/proto/image_classifier_options.proto

@@ -42,7 +42,7 @@ message ImageClassifierOptions {
   // returned.
   optional int32 max_results = 2 [default = -1];
 
-  // Score threshold in [0,1[, overrides the ones provided in the model metadata
+  // Score threshold in [0,1], overrides the ones provided in the model metadata
   // (if any). Results below this value are rejected.
   optional float score_threshold = 3;
 

tensorflow_lite_support/cc/utils/jni_utils.cc

@@ -54,6 +54,22 @@ absl::string_view GetMappedFileBuffer(JNIEnv* env, const jobject& file_buffer) {
       static_cast<size_t>(env->GetDirectBufferCapacity(file_buffer)));
 }
 
+void ThrowException(JNIEnv* env, const char* clazz, const char* fmt, ...) {
+  va_list args;
+  va_start(args, fmt);
+  const size_t max_msg_len = 512;
+  auto* message = static_cast<char*>(malloc(max_msg_len));
+  if (message && (vsnprintf(message, max_msg_len, fmt, args) >= 0)) {
+    env->ThrowNew(env->FindClass(clazz), message);
+  } else {
+    env->ThrowNew(env->FindClass(clazz), "");
+  }
+  if (message) {
+    free(message);
+  }
+  va_end(args);
+}
+
 }  // namespace utils
 }  // namespace support
 }  // namespace tflite

tensorflow_lite_support/cc/utils/jni_utils.h

@@ -28,7 +28,13 @@ namespace tflite {
 namespace support {
 namespace utils {
 
+const char kIllegalArgumentException[] = "java/lang/IllegalArgumentException";
 const char kIllegalStateException[] = "java/lang/IllegalStateException";
+const char kNullPointerException[] = "java/lang/NullPointerException";
+const char kIndexOutOfBoundsException[] = "java/lang/IndexOutOfBoundsException";
+const char kUnsupportedOperationException[] =
+    "java/lang/UnsupportedOperationException";
+const char kAssertionError[] = "java/lang/AssertionError";
 
 // Check if t is nullptr, throw IllegalStateException if it is.
 // Used to verify different types of jobjects are correctly created from jni.
@@ -67,6 +73,9 @@ std::vector<std::string> StringListToVector(JNIEnv* env, jobject list_object);
 
 // Gets a mapped file buffer from a java object representing a file.
 absl::string_view GetMappedFileBuffer(JNIEnv* env, const jobject& file_buffer);
+
+void ThrowException(JNIEnv* env, const char* clazz, const char* fmt, ...);
+
 }  // namespace utils
 }  // namespace support
 }  // namespace tflite

tensorflow_lite_support/java/BUILD

@@ -9,11 +9,12 @@ package(
     licenses = ["notice"],  # Apache 2.0
 )
 
+exports_files(["AndroidManifest.xml"])
+
 android_library(
     name = "tensorflowlite_support",
     srcs = glob(
         ["src/java/org/tensorflow/lite/support/**/*.java"],
-        exclude = ["src/java/org/tensorflow/lite/support/task/**/*.java"],
     ),
     javacopts = JAVACOPTS,
     manifest = "AndroidManifest.xml",