Use a lock to ensure only one segmentation is run at a time
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user