Package ai.djl.modality.cv.translator

Class YoloSegmentationTranslator

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator<DetectedObjects>

ai.djl.modality.cv.translator.ObjectDetectionTranslator

ai.djl.modality.cv.translator.YoloV5Translator

ai.djl.modality.cv.translator.YoloSegmentationTranslator

All Implemented Interfaces:

PostProcessor<DetectedObjects>, PreProcessor<Image>, Translator<Image,DetectedObjects>

public class YoloSegmentationTranslator
extends YoloV5Translator

A translator for Yolov8 instance segmentation models.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	YoloSegmentationTranslator.Builder	The builder for instance segmentation translator.

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

YoloV5Translator.YoloOutputType

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

ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ObjectDetectionTranslator.ObjectDetectionBuilder>

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.ObjectDetectionTranslator

applyRatio, classes, removePadding

Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

height, pipeline, width

Constructor Summary

Constructors

Constructor	Description
YoloSegmentationTranslator(YoloSegmentationTranslator.Builder builder)	Creates the instance segmentation translator from the given builder.

Method Summary

Modifier and Type	Method	Description
static YoloSegmentationTranslator.Builder	builder()	Creates a builder to build a YoloSegmentationTranslator.
static YoloSegmentationTranslator.Builder	builder(Map<String,?> arguments)	Creates a builder to build a YoloSegmentationTranslator with specified arguments.
DetectedObjects	processOutput(TranslatorContext ctx, NDList list)	Processes the output NDList to the corresponding output object.

Methods inherited from class ai.djl.modality.cv.translator.YoloV5Translator

nms, processFromBoxOutput

Methods inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator

prepare

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

getBatchifier, processInput

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

batchProcessInput, batchProcessOutput, getExpansions

Constructor Details

YoloSegmentationTranslator

public YoloSegmentationTranslator(YoloSegmentationTranslator.Builder builder)

Creates the instance segmentation translator from the given builder.

Parameters:

builder - the builder for the translator

Method Details

processOutput

public DetectedObjects processOutput(TranslatorContext ctx,
 NDList list)

Processes the output NDList to the corresponding output object.

Specified by:

processOutput in interface PostProcessor<DetectedObjects>

Overrides:

processOutput in class YoloV5Translator

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference, usually immutable in engines like PyTorch. @see Issue 1774

Returns:

the output object of expected type

builder

public static YoloSegmentationTranslator.Builder builder()

Creates a builder to build a YoloSegmentationTranslator.

Returns:

a new builder

builder

public static YoloSegmentationTranslator.Builder builder(Map<String,?> arguments)

Creates a builder to build a YoloSegmentationTranslator with specified arguments.

Parameters:

arguments - arguments to specify builder options

Returns:

a new builder