ai.djl.modality.cv.translator

Class ObjectDetectionTranslator

java.lang.Object

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

ai.djl.modality.cv.translator.ObjectDetectionTranslator

All Implemented Interfaces:

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

Direct Known Subclasses:

SingleShotDetectionTranslator, YoloTranslator, YoloV5Translator

public abstract class ObjectDetectionTranslator
extends BaseImageTranslator<DetectedObjects>

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

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ObjectDetectionTranslator.ObjectDetectionBuilder> The base builder for the object detection 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

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

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	ObjectDetectionTranslator(ObjectDetectionTranslator.ObjectDetectionBuilder<?> builder) Creates the ObjectDetectionTranslator from the given builder.

Method Summary

Modifier and Type	Method and Description
void	prepare(NDManager manager, Model model) Prepares the translator with the manager and model to use.

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

getBatchifier, getBooleanValue, getFloatValue, getIntValue, getPipeline, getStringValue, 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.PostProcessor

processOutput

Field Detail

threshold

protected float threshold

classes

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

imageWidth

protected double imageWidth

imageHeight

protected double imageHeight

Constructor Detail

ObjectDetectionTranslator

protected ObjectDetectionTranslator(ObjectDetectionTranslator.ObjectDetectionBuilder<?> builder)

Creates the ObjectDetectionTranslator 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

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