outputType : CONFIDENCE_MASK seems weird in Android about mediapipe-samples HOT 6 CLOSED

in setupImageSegmenter() - same thing you did
.setOutputType(ImageSegmenter.ImageSegmenterOptions.OutputType.CONFIDENCE_MASK)

        val mpImage = result.segmentations().first()

        imageSegmenterListener?.onResults(
            ResultBundle(
                ByteBufferExtractor.extract(mpImage).asFloatBuffer(),
                mpImage.width,
                mpImage.height,
                inferenceTime
            )
        )

    fun setResults(
        byteBuffer: FloatBuffer,
        outputWidth: Int,
        outputHeight: Int
    ) {
        val colorLabel = HashSet<Pair<String, Int>>()
        // Create the mask bitmap with colors and the set of detected labels.
        val pixels = IntArray(byteBuffer.capacity())
        for (i in pixels.indices) {
            val confidence = byteBuffer.get()

            val color =
                Color.argb(255 * if( (1.0f - confidence) >= 0.5f ) 0.0f else { confidence }, 0.0f, 0.0f, 128.0f)
            pixels[i] = color
        }
        val image = Bitmap.createBitmap(
            pixels,
            outputWidth,
            outputHeight,
            Bitmap.Config.ARGB_8888
        )

        val scaleFactor = when (runningMode) {
            RunningMode.IMAGE,
            RunningMode.VIDEO -> {
                min(width * 1f / outputWidth, height * 1f / outputHeight)
            }
            RunningMode.LIVE_STREAM -> {
                // PreviewView is in FILL_START mode. So we need to scale up the
                // landmarks to match with the size that the captured images will be
                // displayed.
                max(width * 1f / outputWidth, height * 1f / outputHeight)
            }
        }

        val scaleWidth = (outputWidth * scaleFactor).toInt()
        val scaleHeight = (outputHeight * scaleFactor).toInt()

        scaleBitmap = Bitmap.createScaledBitmap(
            image, scaleWidth, scaleHeight, false
        )
        invalidate()
        listener?.onLabels(colorLabel.toList())
    }