Add function to check RKNN quantization in JNI (#26)

* add is quantized check

* add non quant model to check

* actually load the lib lol

* fix issues hopefully?

* im hoping this rebase works

* resolve merge conflicts

* add non quant model to check

* Add linting workflow (#27)

# Conflicts:
#	src/main/java/org/photonvision/rknn/RknnJNI.java
#	src/main/native/include/yolo_common.hpp
#	src/test/java/org/photonvision/rknn/RknnTest.java

* fix member ordering

Author: boomermath

coralAlgaeModelNoQuant-640-640-yolov8s.rknn

@@ -74,7 +74,7 @@ public boolean equals(Object obj) {
      * @param modelPath Absolute path to the model on disk
      * @param numClasses How many classes. MUST MATCH or native code segfaults
      * @param modelVer Which model is being used. Detections will be incorrect if not set to
-     *     corrresponding model.
+     *     corresponding model.
      * @return Pointer to the detector in native memory
      */
     public static native long create(String modelPath, int numClasses, int modelVer, int coreNum);
@@ -89,11 +89,11 @@ public boolean equals(Object obj) {
      */
     public static native int setCoreMask(long ptr, int desiredCore);
 
-    /** Delete all native resources assocated with a detector */
+    /** Delete all native resources associated with a detector */
     public static native long destroy(long ptr);
 
     /**
-     * Run detction
+     * Run detection
      *
      * @param detectorPtr Pointer to detector created above
      * @param imagePtr Pointer to a cv::Mat input image
@@ -102,4 +102,12 @@ public boolean equals(Object obj) {
      */
     public static native RknnResult[] detect(
             long detectorPtr, long imagePtr, double nmsThresh, double boxThresh);
+
+    /**
+     * Check if a model is quantized (int8)
+     *
+     * @param detectorPtr Pointer to a detected created above
+     * @return true if quantized, false if not
+     */
+    public static native boolean isQuantized(long detectorPtr);
 }

src/main/java/org/photonvision/rknn/RknnJNI.java

@@ -143,4 +143,16 @@ Java_org_photonvision_rknn_RknnJNI_detect
   return jarr;
 }
 
+/*
+ * Class:     org_photonvision_rknn_RknnJNI
+ * Method:    isQuantized
+ * Signature: (J)Z
+ */
+JNIEXPORT jboolean JNICALL
+Java_org_photonvision_rknn_RknnJNI_isQuantized
+  (JNIEnv *env, jclass, jlong detector_)
+{
+  YoloModel *yolo = reinterpret_cast<YoloModel *>(detector_);
+  return yolo->is_quant ? JNI_TRUE : JNI_FALSE;
+}
 } // extern "C"

src/main/native/cpp/rknn_jni.cpp

@@ -57,6 +57,14 @@ JNIEXPORT void JNICALL Java_org_photonvision_rknn_RknnJNI_destroy(JNIEnv *,
 JNIEXPORT jobjectArray JNICALL Java_org_photonvision_rknn_RknnJNI_detect(
     JNIEnv *, jclass, jlong, jlong, jdouble, jdouble);
 
+/*
+ * Class:     org_photonvision_rknn_RknnJNI
+ * Method:    isQuantized
+ * Signature: (J)Z
+ */
+JNIEXPORT jboolean JNICALL
+Java_org_photonvision_rknn_RknnJNI_isQuantized(JNIEnv *, jclass, jlong);
+
 #ifdef __cplusplus
 } // extern "C"
 #endif

src/main/native/cpp/rknn_jni.h

@@ -44,6 +44,8 @@ class YoloModel {
 
   ~YoloModel();
 
+  bool is_quant;
+
 protected:
   virtual detect_result_group_t postProcess(std::vector<rknn_output> output,
                                             DetectionFilterParams params,
@@ -63,7 +65,6 @@ class YoloModel {
   rknn_input_output_num io_num;
   std::vector<rknn_tensor_attr> input_attrs;
   std::vector<rknn_tensor_attr> output_attrs;
-  bool is_quant;
 
   rknn_input inputs[1];
 };

src/main/native/include/yolo_common.hpp

@@ -17,6 +17,8 @@
 
 package org.photonvision.rknn;
 
+import static org.junit.jupiter.api.Assertions.*;
+
 import edu.wpi.first.util.CombinedRuntimeLoader;
 import java.io.IOException;
 import java.util.Arrays;
@@ -68,4 +70,32 @@ public void testBasicBlobs() {
         System.out.println("Killing detector");
         RknnJNI.destroy(ptr);
     }
+
+    private boolean checkQuant(String path, int numClasses, ModelVersion modelVer) {
+        long ptr = RknnJNI.create(path, numClasses, modelVer.ordinal(), 210);
+
+        if (ptr <= 0) {
+            throw new RuntimeException("Failed to create model");
+        }
+
+        boolean isQuantized = RknnJNI.isQuantized(ptr);
+
+        RknnJNI.destroy(ptr);
+
+        return isQuantized;
+    }
+
+    @Test
+    public void testQuantizationCheck() {
+        System.out.println("Loading rknn-jni");
+        System.load("/home/coolpi/rknn_jni/cmake_build/librknn_jni.so");
+
+        assertTrue(
+                checkQuant("/home/coolpi/rknn_jni/note-640-640-yolov5s.rknn", 1, ModelVersion.YOLO_V5));
+        assertFalse(
+                checkQuant(
+                        "/home/coolpi/rknn_jni/coralAlgaeModelNoQuant-640-640-yolov8s.rknn",
+                        2,
+                        ModelVersion.YOLO_V8));
+    }
 }