App/Vision: Update the ImageAnalyzer class to support frame-skipping

This patch updates the ImageAnalyzer class, which is an abstraction layer of the concrete image analysis implementation, to support condition-based frame-skipping. Signed-off-by: Wook Song <[email protected]>
nnstreamer · Nov 14, 2024 · e770a07 · e770a07
1 parent c11c2a2
commit e770a07
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/...ence_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/ImageAnalyzer.kt b/...ence_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/data/ImageAnalyzer.kt
@@ -6,13 +6,20 @@ import androidx.camera.core.ImageProxy
 
 class ImageAnalyzer(
     private val classifier: MobilenetClassifier,
+    private val frameSkipRate: Int = 30,
+    private val shouldAnalyze: (ImageProxy, Int) -> Boolean = { _, frameCount -> frameCount % frameSkipRate == 0 }
 ) : ImageAnalysis.Analyzer {
+    private var frameSkipCounter = 0
+
     override fun analyze(imageProxy: ImageProxy) {
-        val rotationDegrees = imageProxy.imageInfo.rotationDegrees
-        val bitmap = imageProxy.toBitmap()
+        if (shouldAnalyze(imageProxy, frameSkipCounter)) {
+            val rotationDegrees = imageProxy.imageInfo.rotationDegrees
+            val bitmap = imageProxy.toBitmap()
 
-        classifier.classify(bitmap, rotationDegrees)
+            classifier.classify(bitmap, rotationDegrees)
+        }
 
+        frameSkipCounter = (frameSkipCounter + 1) % frameSkipRate
         imageProxy.close()
     }
 }