ai.djl.modality.cv.translator

Class YoloTranslator

java.lang.Object

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

ai.djl.modality.cv.translator.ObjectDetectionTranslator

ai.djl.modality.cv.translator.YoloTranslator

All Implemented Interfaces:

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

public class YoloTranslator
extends ObjectDetectionTranslator

A translator for yolo models.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	YoloTranslator.Builder The builder for YoloTranslator.

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

ObjectDetectionTranslator.BaseBuilder<T extends ObjectDetectionTranslator.BaseBuilder>

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

BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.SynsetLoader

Field Summary

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

classes, imageHeight, imageWidth, threshold

Constructor Summary

Constructors

Constructor and Description
YoloTranslator(YoloTranslator.Builder builder) Constructs an ImageTranslator with the provided builder.

Method Summary

Modifier and Type	Method and Description
static YoloTranslator.Builder	builder() Creates a builder to build a YoloTranslator.
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, getPipeline, processInput

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

YoloTranslator

public YoloTranslator(YoloTranslator.Builder builder)

Constructs an ImageTranslator with the provided builder.

Parameters:

builder - the data to build with

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

Creates a builder to build a YoloTranslator.

Returns:

a new builder