cyy/Fairscan_cyy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix computation of the number of objects that are detected

Browse Source
This commit is contained in:
Pierre-Yves Nicolas
2025-05-27 08:10:26 +02:00
parent f8f1f6bab2
commit 95bd3bd823
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
app/src/main/java/org/mydomain/myscan/MainViewModel.kt
View File

@@ -32,7 +32,7 @@ class MainViewModel(private val imageSegmentationService: ImageSegmentationServi
imageSegmentationService.initialize()
imageSegmentationService.segmentation
.filterNotNull()
.map { UiState("Found ${it.segmentation.masks.size} objects!", it.inferenceTime) }
.map { UiState("Found ${numberOfObjectsDetected(it.segmentation)} objects!", it.inferenceTime) }
.collect {
Log.d("MyScan", "New UIstate ${it}")
_uiState.value = it
@@ -40,6 +40,15 @@ class MainViewModel(private val imageSegmentationService: ImageSegmentationServi
}
}
fun numberOfObjectsDetected(segmentation: ImageSegmentationService.Segmentation) : Int {
val tensor = segmentation.masks[0];
val buffer = tensor.buffer
val uniqueValues = HashSet<Int>()
for (i in 0..tensor.width * tensor.height - 1) {
uniqueValues.add(buffer[i].toInt())
}
return uniqueValues.size - 1;
}
fun segment(imageProxy: ImageProxy) {
Log.d("MyScan", "MainViewModel.Calling segment")