amazon-chime-sdk/com.amazonaws.services.chime.sdk.meetings.audiovideo.video.backgroundfilter/SegmentationProcessor

SegmentationProcessor

class SegmentationProcessor(val context: Context)

SegmentationProcessor predicts foreground mask for an image.

Constructors

SegmentationProcessor

constructor(context: Context)

Types

object Companion

Properties

val context: Context

lateinit var modelState: ModelState

Functions

fun getInputBuffer(): ByteBuffer

Retrieve input image on which segmentation can be applied.

getOutputBuffer

fun getOutputBuffer(): ByteBuffer

Retrieve segmented image mask.

fun initialize(width: Int, height: Int, modelShape: ModelShape)

Initialize and load the tensorflow model.

fun predict(): PredictResult

Predicts the foreground on the input image.