ai.djl.modality.cv.translator

Class SingleShotDetectionTranslator

java.lang.Object

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

ai.djl.modality.cv.translator.SingleShotDetectionTranslator

All Implemented Interfaces:

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

public class SingleShotDetectionTranslator
extends ImageTranslator<DetectedObjects>

A ImageTranslator that post-process the NDArray into DetectedObjects with boundaries.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SingleShotDetectionTranslator.Builder The builder for SSD translator.

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

ImageTranslator.BaseBuilder<T extends ImageTranslator.BaseBuilder>

Field Summary

Fields

Modifier and Type	Field and Description
protected java.util.List<java.lang.String>	classes
protected java.lang.String	synsetArtifactName
protected float	threshold

Constructor Summary

Constructors

Constructor and Description
SingleShotDetectionTranslator(SingleShotDetectionTranslator.Builder builder) Creates the SSD translator from the given builder.

Method Summary

Modifier and Type	Method and Description
static SingleShotDetectionTranslator.Builder	builder() Creates a builder to build a SingleShotDetectionTranslator.
void	prepare(NDManager manager, Model model) Prepares the translator with the manager and model to use.
DetectedObjects	processOutput(TranslatorContext ctx, NDList list) Processes the output NDList to the corresponding output object.

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

getPipeline, 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

getBatchifier

Field Detail

threshold

protected float threshold

synsetArtifactName

protected java.lang.String synsetArtifactName

classes

protected java.util.List<java.lang.String> classes

Constructor Detail

SingleShotDetectionTranslator

public SingleShotDetectionTranslator(SingleShotDetectionTranslator.Builder builder)

Creates the SSD translator from the given builder.

Parameters:

builder - the builder for the translator

Method Detail

prepare

public void prepare(NDManager manager,
                    Model model)
             throws java.io.IOException

Description copied from interface: Translator

Prepares the translator with the manager and model to use.

Parameters:

manager - the manager for the translator

model - the model to translate for

Throws:

java.io.IOException - if there is an error reading inputs for preparing the translator

processOutput

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

Processes the output NDList to the corresponding output object.

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 SingleShotDetectionTranslator.Builder builder()

Creates a builder to build a SingleShotDetectionTranslator.

Returns:

a new builder