114 set minimal inference time (#116)

cvlab-ai · Jun 2, 2023 · 5f66111 · 5f66111
1 parent b25044d
commit 5f66111
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/...nknotesApp/app/src/client/java/pg/eti/project/polishbanknotes/fragments/CameraFragment.kt b/...nknotesApp/app/src/client/java/pg/eti/project/polishbanknotes/fragments/CameraFragment.kt
@@ -46,16 +46,13 @@ import pg.eti.project.polishbanknotes.sensors.TorchManager
 import pg.eti.project.polishbanknotes.settings_management.LabelManager
 import java.util.concurrent.ExecutorService
 import java.util.concurrent.Executors
+import kotlin.math.floor
 
 /**
  * Number of last results that will be considered for choosing final label
  * (most occurrences in last NUMBER_OF_LAST_RESULTS)
  */
-// TODO optimization: auto select inference counter for older devices.
-//  on Xiaomi Redmi 6A the app is slow and inference is giving message even if not
-//  pointing on banknote.
-// TODO question: is it needed?
-const val NUMBER_OF_LAST_RESULTS = 5
+
 const val SLIGHTLY_MORE_MILLIS = 500L
 
 class CameraFragment : Fragment(), ImageClassifierHelper.ClassifierListener {
@@ -96,6 +93,8 @@ class CameraFragment : Fragment(), ImageClassifierHelper.ClassifierListener {
     private var classificationActive = true
     private var inferenceMillisCounter: Long = 0L
     private var lastLabels = mutableListOf<String?>()
+    private var inferenceCounter = 0
+    private var NUMBER_OF_LAST_RESULTS = 5
     private lateinit var beeper: Beeper
 
     private lateinit var haptizer: Haptizer
@@ -355,6 +354,16 @@ class CameraFragment : Fragment(), ImageClassifierHelper.ClassifierListener {
             classificationResultsAdapter.updateResults(results)
             classificationResultsAdapter.notifyDataSetChanged()
 
+            if(inferenceCounter <= 4) {
+                if(inferenceCounter == 4) {
+                    val lastResultsCalculation: Int = floor(500 / inferenceTime.toDouble()).toInt()
+
+                    if(lastResultsCalculation > 5)
+                        NUMBER_OF_LAST_RESULTS = lastResultsCalculation
+                }
+                inferenceCounter++
+            }
+
             val result: String? = if(results?.isEmpty() == true || results!![0].categories.isEmpty()){
                 null
             }else{