Use a lock to ensure only one segmentation is run at a time

This commit is contained in:
Pierre-Yves Nicolas
2025-06-19 17:52:03 +02:00
parent 0c5a219783
commit 453a8b3fb0
2 changed files with 14 additions and 7 deletions

View File

@@ -21,6 +21,7 @@ import android.util.Log
import androidx.test.core.app.ApplicationProvider
import androidx.test.platform.app.InstrumentationRegistry
import androidx.test.ext.junit.runners.AndroidJUnit4
import kotlinx.coroutines.runBlocking
import org.junit.Test
import org.junit.runner.RunWith
@@ -47,7 +48,9 @@ class DocumentDetectionTest {
val bitmap = BitmapFactory.decodeStream(inputStream)
var outputBitmap: Bitmap? = null
val segmentationResult = segmentationService.runSegmentationAndReturn(bitmap, 0)
val segmentationResult = runBlocking {
segmentationService.runSegmentationAndReturn(bitmap, 0)
}
if (segmentationResult != null) {
val mask = segmentationResult.segmentation.toBinaryMask()
val quad = detectDocumentQuad(mask)