@@ -8,16 +8,13 @@ import com.google.mediapipe.tasks.core.Delegate
88import com.google.mediapipe.tasks.vision.core.RunningMode
99import com.google.mediapipe.tasks.vision.imagesegmenter.ImageSegmenter
1010import com.google.mediapipe.tasks.vision.imagesegmenter.ImageSegmenterResult
11- import java.nio.ByteBuffer
1211
1312class ImageSegmenterHelper (
1413 var currentDelegate : Int = DELEGATE_CPU ,
1514 var currentModel : Int = MODEL_DEEPLABV3 ,
1615 val context : Context ,
1716) {
1817
19- // For this example this needs to be a var so it can be reset on changes. If the Imagesegmenter
20- // will not change, a lazy val would be preferable.
2118 private var imagesegmenter: ImageSegmenter ? = null
2219
2320 init {
@@ -79,21 +76,10 @@ class ImageSegmenterHelper(
7976 }
8077 }
8178
82- // Runs image segmentation on single image and
83- // returns the results asynchronously to the caller.
8479 fun segmentImageFile (mpImage : MPImage ): ImageSegmenterResult ? {
8580 return imagesegmenter?.segment(mpImage)
8681 }
8782
88- // Wraps results from inference, the time it takes for inference to be
89- // performed.
90- data class ResultBundle (
91- val results : ByteBuffer ,
92- val width : Int ,
93- val height : Int ,
94- val inferenceTime : Long ,
95- )
96-
9783 companion object {
9884 const val DELEGATE_CPU = 0
9985 const val DELEGATE_GPU = 1
0 commit comments