Class PtSsdTranslator
- java.lang.Object
-
- ai.djl.modality.cv.translator.BaseImageTranslator<ai.djl.modality.cv.output.DetectedObjects>
-
- ai.djl.modality.cv.translator.ObjectDetectionTranslator
-
- ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator
-
- All Implemented Interfaces:
ai.djl.translate.PostProcessor<ai.djl.modality.cv.output.DetectedObjects>
,ai.djl.translate.PreProcessor<ai.djl.modality.cv.Image>
,ai.djl.translate.Translator<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>
public class PtSsdTranslator extends ai.djl.modality.cv.translator.ObjectDetectionTranslator
APtSsdTranslator
that post-process theNDArray
intoDetectedObjects
with boundaries. Reference implementation: SSD.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
PtSsdTranslator.Builder
The builder for SSD translator.-
Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator
ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder>
-
Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<T extends ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder>, ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder<T extends ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder>, ai.djl.modality.cv.translator.BaseImageTranslator.SynsetLoader
-
-
Constructor Summary
Constructors Modifier Constructor Description protected
PtSsdTranslator(PtSsdTranslator.Builder builder)
Creates the SSD translator from the given builder.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static PtSsdTranslator.Builder
builder()
Creates a builder to build aPtSSDTranslatorBuilder
.static PtSsdTranslator.Builder
builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aPtSSDTranslatorBuilder
with specified arguments.void
prepare(ai.djl.translate.TranslatorContext ctx)
ai.djl.modality.cv.output.DetectedObjects
processOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
-
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
getBatchifier, processInput
-
-
-
-
Constructor Detail
-
PtSsdTranslator
protected PtSsdTranslator(PtSsdTranslator.Builder builder)
Creates the SSD translator from the given builder.- Parameters:
builder
- the builder for the translator
-
-
Method Detail
-
prepare
public void prepare(ai.djl.translate.TranslatorContext ctx) throws java.lang.Exception
- Specified by:
prepare
in interfaceai.djl.translate.Translator<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>
- Overrides:
prepare
in classai.djl.modality.cv.translator.ObjectDetectionTranslator
- Throws:
java.lang.Exception
-
processOutput
public ai.djl.modality.cv.output.DetectedObjects processOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
-
builder
public static PtSsdTranslator.Builder builder()
Creates a builder to build aPtSSDTranslatorBuilder
.- Returns:
- a new builder
-
builder
public static PtSsdTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aPtSSDTranslatorBuilder
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-
-