ai.djl.modality.cv

Class InstanceSegmentationTranslator

java.lang.Object

ai.djl.modality.cv.ImageTranslator<DetectedObjects>

ai.djl.modality.cv.InstanceSegmentationTranslator

All Implemented Interfaces:

PostProcessor<DetectedObjects>, PreProcessor<java.awt.image.BufferedImage>, Transform, Translator<java.awt.image.BufferedImage,DetectedObjects>

public class InstanceSegmentationTranslator
extends ImageTranslator<DetectedObjects>
implements Transform

A ImageTranslator that post-process the NDArray into DetectedObjects with boundaries at the detailed pixel level.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	InstanceSegmentationTranslator.Builder The builder for Instance Segmentation translator.

Nested classes/interfaces inherited from class ai.djl.modality.cv.ImageTranslator

ImageTranslator.BaseBuilder<T extends ImageTranslator.BaseBuilder>

Constructor Summary

Constructors

Constructor and Description
InstanceSegmentationTranslator(InstanceSegmentationTranslator.Builder builder) Creates the Instance Segmentation translator from the given builder.

Method Summary

Modifier and Type	Method and Description
static InstanceSegmentationTranslator.Builder	builder() Creates a builder to build a InstanceSegmentationTranslator.
NDList	processInput(TranslatorContext ctx, java.awt.image.BufferedImage image) Processes the BufferedImage input and converts it to NDList.
DetectedObjects	processOutput(TranslatorContext ctx, NDList list) Processes the output NDList to the corresponding output object.
NDArray	transform(NDArray array) Applies the Transform to the given NDArray.

Methods inherited from class ai.djl.modality.cv.ImageTranslator

getPipeline

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.translate.Translator

getBatchifier

Constructor Detail

InstanceSegmentationTranslator

public InstanceSegmentationTranslator(InstanceSegmentationTranslator.Builder builder)

Creates the Instance Segmentation translator from the given builder.

Parameters:

builder - the builder for the translator

Method Detail

transform

public NDArray transform(NDArray array)

Applies the Transform to the given NDArray.

Specified by:

transform in interface Transform

Parameters:

array - the NDArray on which the Transform is applied

Returns:

the output of the Transform

processInput

public NDList processInput(TranslatorContext ctx,
                           java.awt.image.BufferedImage image)

Processes the BufferedImage input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<java.awt.image.BufferedImage>

Overrides:

processInput in class ImageTranslator<DetectedObjects>

Parameters:

ctx - the toolkit that helps create the input NDArray

image - the BufferedImage input

Returns:

a NDList

processOutput

public DetectedObjects processOutput(TranslatorContext ctx,
                                     NDList list)
                              throws java.io.IOException

Processes the output NDList to the corresponding output object.

Specified by:

processOutput in interface PostProcessor<DetectedObjects>

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference

Returns:

the output object of expected type

Throws:

java.io.IOException

builder

public static InstanceSegmentationTranslator.Builder builder()

Creates a builder to build a InstanceSegmentationTranslator.

Returns:

a new builder