Fix the JNI local ref overflow issue.

PiperOrigin-RevId: 351906708
tensorflow · Jan 15, 2021 · 6ccadfc · 6ccadfc
1 parent a63d02d
commit 6ccadfc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tensorflow_lite_support/java/src/native/task/vision/jni_utils.cc b/tensorflow_lite_support/java/src/native/task/vision/jni_utils.cc
@@ -50,6 +50,9 @@ jobject ConvertToCategory(JNIEnv* env, const Class& classification) {
   jobject jcategory =
       env->CallStaticObjectMethod(category_class, category_create, label,
                                   display_name, classification.score());
+  env->DeleteLocalRef(category_class);
+  env->DeleteLocalRef(label);
+  env->DeleteLocalRef(display_name);
   return jcategory;
 }