Package ai.djl.modality.cv.translator
Class InstanceSegmentationTranslator
- java.lang.Object
-
- ai.djl.modality.cv.translator.BaseImageTranslator<DetectedObjects>
-
- ai.djl.modality.cv.translator.InstanceSegmentationTranslator
-
- All Implemented Interfaces:
PostProcessor<DetectedObjects>
,PreProcessor<Image>
,Translator<Image,DetectedObjects>
public class InstanceSegmentationTranslator extends BaseImageTranslator<DetectedObjects>
ABaseImageTranslator
that post-process theNDArray
intoDetectedObjects
with boundaries at the detailed pixel level.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
InstanceSegmentationTranslator.Builder
The builder for Instance Segmentation translator.-
Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.SynsetLoader
-
-
Field Summary
-
Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
pipeline
-
-
Constructor Summary
Constructors Constructor Description InstanceSegmentationTranslator(InstanceSegmentationTranslator.Builder builder)
Creates the Instance Segmentation translator from the given builder.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static InstanceSegmentationTranslator.Builder
builder()
Creates a builder to build aInstanceSegmentationTranslator
.static InstanceSegmentationTranslator.Builder
builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aInstanceSegmentationTranslator
with specified arguments.void
prepare(TranslatorContext ctx)
Prepares the translator with the manager and model to use.NDList
processInput(TranslatorContext ctx, Image image)
Processes the input and converts it to NDList.DetectedObjects
processOutput(TranslatorContext ctx, NDList list)
Processes the output NDList to the corresponding output object.-
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
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
-
prepare
public void prepare(TranslatorContext ctx) throws java.io.IOException
Prepares the translator with the manager and model to use.- Parameters:
ctx
- the context for thePredictor
.- Throws:
java.io.IOException
-
processInput
public NDList processInput(TranslatorContext ctx, Image image)
Processes the input and converts it to NDList.- Specified by:
processInput
in interfacePreProcessor<Image>
- Overrides:
processInput
in classBaseImageTranslator<DetectedObjects>
- Parameters:
ctx
- the toolkit for creating the input NDArrayimage
- the input object- Returns:
- the
NDList
after pre-processing
-
processOutput
public DetectedObjects processOutput(TranslatorContext ctx, NDList list)
Processes the output NDList to the corresponding output object.- Parameters:
ctx
- the toolkit used for post-processinglist
- the output NDList after inference- Returns:
- the output object of expected type
-
builder
public static InstanceSegmentationTranslator.Builder builder()
Creates a builder to build aInstanceSegmentationTranslator
.- Returns:
- a new builder
-
builder
public static InstanceSegmentationTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aInstanceSegmentationTranslator
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-
-