Package ai.djl.modality.cv.translator

Class YoloV8Translator

java.lang.Object

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

ai.djl.modality.cv.translator.ObjectDetectionTranslator

ai.djl.modality.cv.translator.YoloV5Translator

ai.djl.modality.cv.translator.YoloV8Translator

All Implemented Interfaces:

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

public class YoloV8Translator
extends YoloV5Translator

A translator for YoloV8 models. This was tested with ONNX exported Yolo models. For details check here

Nested Class Summary

Nested Classes

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

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

YoloV5Translator.YoloOutputType

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

applyRatio, classes, removePadding, threshold

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

height, pipeline, width

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	YoloV8Translator(YoloV8Translator.Builder builder)	Constructs an ImageTranslator with the provided builder.

Method Summary

Modifier and Type	Method	Description
static YoloV8Translator.Builder	builder()	Creates a builder to build a YoloV8Translator with specified arguments.
static YoloV8Translator.Builder	builder(Map<String,?> arguments)	Creates a builder to build a YoloV8Translator with specified arguments.
protected DetectedObjects	processFromBoxOutput(int imageWidth, int imageHeight, NDList list)

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

nms, processOutput

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

Methods inherited from interface ai.djl.translate.Translator

batchProcessInput, batchProcessOutput, getExpansions

Constructor Details

YoloV8Translator

protected YoloV8Translator(YoloV8Translator.Builder builder)

Constructs an ImageTranslator with the provided builder.

Parameters:

builder - the data to build with

Method Details

builder

public static YoloV8Translator.Builder builder()

Creates a builder to build a YoloV8Translator with specified arguments.

Returns:

a new builder

builder

public static YoloV8Translator.Builder builder(Map<String,?> arguments)

Creates a builder to build a YoloV8Translator with specified arguments.

Parameters:

arguments - arguments to specify builder options

Returns:

a new builder

processFromBoxOutput

protected DetectedObjects processFromBoxOutput(int imageWidth,
 int imageHeight,
 NDList list)

Overrides:

processFromBoxOutput in class YoloV5Translator