ai.djl.modality.cv.translator

Class SingleShotDetectionTranslator

java.lang.Object

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

ai.djl.modality.cv.translator.ObjectDetectionTranslator

ai.djl.modality.cv.translator.SingleShotDetectionTranslator

All Implemented Interfaces:

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

public class SingleShotDetectionTranslator
extends ObjectDetectionTranslator

A BaseImageTranslator 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.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

classes, imageHeight, imageWidth, threshold

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

pipeline

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.
static SingleShotDetectionTranslator.Builder	builder(java.util.Map<java.lang.String,?> arguments) Creates a builder to build a SingleShotDetectionTranslator 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.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

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

processOutput

public DetectedObjects processOutput(TranslatorContext ctx,
                                     NDList list)

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

builder

public static SingleShotDetectionTranslator.Builder builder()

Creates a builder to build a SingleShotDetectionTranslator.

Returns:

a new builder

builder

public static SingleShotDetectionTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)

Creates a builder to build a SingleShotDetectionTranslator with specified arguments.

Parameters:

arguments - arguments to specify builder options

Returns:

a new builder